Add check_requirements() (#1853)

* Add check_requirements()

* add import

* parameterize filename

* add to detect, test
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-01-06 16:35:40 -08:00
1 parent 135ec5c5ce
commit e77c77f580
4 files changed
+16 -5

No files matched your search

+3 -2
View File
@@ -9,8 +9,8 @@ from numpy import random
from models.experimental import attempt_load
from utils.datasets import LoadStreams, LoadImages
from utils.general import check_img_size, non_max_suppression, apply_classifier, scale_coords, xyxy2xywh, \
strip_optimizer, set_logging, increment_path
from utils.general import check_img_size, check_requirements, non_max_suppression, apply_classifier, scale_coords, \
xyxy2xywh, strip_optimizer, set_logging, increment_path
from utils.plots import plot_one_box
from utils.torch_utils import select_device, load_classifier, time_synchronized
@@ -162,6 +162,7 @@ if __name__ == '__main__':
parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
opt = parser.parse_args()
print(opt)
check_requirements()
with torch.no_grad():
if opt.update: # update all models (to fix SourceChangeWarning)