Browse Source
check_requirements() exclude pycocotools, thop (#2571)
Exclude non-critical packages from dependency checks in detect.py. pycocotools and thop in particular are not required for inference. Issue first raised in https://github.com/ultralytics/yolov5/issues/1944 and also raised in https://github.com/ultralytics/yolov5/discussions/25565.0
Glenn Jocher
GitHub
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
detect.py
View File
| parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment') | parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment') | ||||
| opt = parser.parse_args() | opt = parser.parse_args() | ||||
| print(opt) | print(opt) | ||||
| check_requirements() | |||||
| check_requirements(exclude=('pycocotools', 'thop')) | |||||
| with torch.no_grad(): | with torch.no_grad(): | ||||
| if opt.update: # update all models (to fix SourceChangeWarning) | if opt.update: # update all models (to fix SourceChangeWarning) |
Loading…