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
@@ -175,7 +175,7 @@ if __name__ == '__main__':
parser.add_argument('--half', action='store_true', help='use FP16 half-precision inference')
opt = parser.parse_args()
print(opt)
check_requirements(exclude=('tensorboard', 'pycocotools', 'thop'))
check_requirements(exclude=('tensorboard', 'thop'))
if opt.update: # update all models (to fix SourceChangeWarning)
for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']: