Spaces:

tidalove
/

yolox

Sleeping

ruinmessi commited on Aug 2, 2021

Commit

5b6dfa8

1 Parent(s): 78a233c

Fix coco eval bug when yolox is not installed correctly (#312)

Files changed (1) hide show

yolox/evaluators/coco_evaluator.py CHANGED Viewed

@@ -206,7 +206,7 @@ class COCOEvaluator:
             try:
                 from yolox.layers import COCOeval_opt as COCOeval
             except ImportError:
-                from pycocotools import cocoeval as COCOeval
                 logger.warning("Use standard COCOeval.")

             try:
                 from yolox.layers import COCOeval_opt as COCOeval
             except ImportError:
+                from pycocotools.cocoeval import COCOeval
                 logger.warning("Use standard COCOeval.")