diff --git a/rlepose/models/layers/__init__.py b/rlepose/models/layers/__init__.py new file mode 100644 index 0000000..8b13789 --- /dev/null +++ b/rlepose/models/layers/__init__.py @@ -0,0 +1 @@ + diff --git a/setup.py b/setup.py index 25de1d0..57c377d 100644 --- a/setup.py +++ b/setup.py @@ -133,12 +133,12 @@ def get_ext_modules(): def get_install_requires(): install_requires = [ - 'six', 'terminaltables', 'scipy==1.1.0', + 'six', 'terminaltables', 'scipy>=1.1.0', 'opencv-python', 'matplotlib', 'visdom', 'tqdm', 'tensorboardx', 'easydict', 'pyyaml', 'torch>=1.1.0', 'torchvision>=0.3.0', - 'munkres', 'timm==0.1.20', 'natsort' + 'munkres', 'timm>=0.1.20', 'natsort' ] return install_requires