Update some requirements, depend on yet to be released timm v0.2.2, update pycocotools dep to 2.0.2

Author: rwightman

effdet/data/parsers/parser_open_images.py

@@ -3,7 +3,6 @@
 Copyright 2020 Ross Wightman
 """
 import numpy as np
-import pandas as pd
 import os
 import logging
 
@@ -44,6 +43,8 @@ def _load_annotations(
             masks_filename: str,
             bbox_filename: str,
     ):
+        import pandas as pd  # For now, blow up on pandas req only when trying to load open images anno
+
         _logger.info('Loading categories...')
         classes_df = pd.read_csv(categories_filename, header=None)
         self.cat_ids = classes_df[0].tolist()

requirements.txt

@@ -1,15 +1,15 @@
 torch>=1.4.0
-timm>=0.1.28
+timm>=0.2.2
 torchvision
-pycocotools
 pyyaml
+numpy
+
+# an update version that fixes some bugs, incl issues with numpy >= 1.18
+pycocotools>=2.0.2
 
 # better to manually install pillow-simd
 Pillow
 
-# conflict between pycocotools and numpy 1.18.+
-numpy<=1.17.5
-
 # pycocotools needs this but doesn't have dep
 matplotlib
 

setup.py

@@ -42,6 +42,6 @@
     # Note that this is a string of words separated by whitespace, not a list.
     keywords='pytorch pretrained efficientdet efficientnet bifpn object detection',
     packages=find_packages(exclude=['data']),
-    install_requires=['torch >= 1.4', 'torchvision', 'timm >= 0.1.28'],
+    install_requires=['torch >= 1.4', 'torchvision', 'timm >= 0.2.2'],
     python_requires='>=3.6',
 )