On-demand pycocotools pip install (#3547)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-06-09 11:22:21 +02:00
1 parent 1b5edb6f8e
commit ef0b5c9d29
5 files changed
+6 -5

No files matched your search

+1 -1
View File
@@ -31,7 +31,7 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
from utils.torch_utils import select_device
check_requirements(requirements=Path(__file__).parent / 'requirements.txt',
exclude=('tensorboard', 'pycocotools', 'thop', 'opencv-python'))
exclude=('tensorboard', 'thop', 'opencv-python'))
set_logging(verbose=verbose)
fname = Path(name).with_suffix('.pt') # checkpoint filename