fix requirement config for flowers dataset

8 years ago · fc5972ba2c
parent 80642bee00
commit fc5972ba2c
1 changed files with 2 additions and 1 deletions
--- a/python/setup.py.in
+++ b/python/setup.py.in
@ -15,7 +15,8 @@ setup_requires=["requests",
                "protobuf==3.1",
                "recordio",
                "matplotlib",
-                "rarfile"]
+                "rarfile",
+                "scipy>=0.19.0"]

 if '${CMAKE_SYSTEM_PROCESSOR}' not in ['arm', 'armv7-a', 'aarch64']:
    setup_requires+=["opencv-python"]