diff --git a/setup.py b/setup.py index d8c6fecda..4adcf9618 100644 --- a/setup.py +++ b/setup.py @@ -53,7 +53,7 @@ def run(self, *args, **kwargs): maintainer_email = 'h.gaiser@fizyr.com', 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',