Update setup.py

broadinstitute/keras-resnet#57
kazushi-fa · Jul 5, 2019 · 1b3e011 · 1b3e011
1 parent b843831
commit 1b3e011
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -53,7 +53,7 @@ def run(self, *args, **kwargs):
     maintainer_email = '[email protected]',
     cmdclass         = {'build_ext': BuildExtension},
     packages         = setuptools.find_packages(),
-    install_requires = ['keras', 'keras-resnet', 'six', 'scipy', 'cython', 'Pillow', 'opencv-python', 'progressbar2'],
+    install_requires = ['keras', 'keras-resnet==0.1.0', 'six', 'scipy', 'cython', 'Pillow', 'opencv-python', 'progressbar2'],
     entry_points     = {
         'console_scripts': [
             'retinanet-train=keras_retinanet.bin.train:main',