From 5323ad224d90680da9cdb7fd8b82089750b7252e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 1 Jul 2020 16:15:25 -0700 Subject: [PATCH] export.py update --- detect.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/detect.py b/detect.py index 93faf6d..2650c20 100644 --- a/detect.py +++ b/detect.py @@ -156,9 +156,9 @@ if __name__ == '__main__': print(opt) with torch.no_grad(): - # detect() + detect() - # Update all models - for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']: - detect() - create_pretrained(opt.weights, opt.weights) + # # Update all models + # for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']: + # detect() + # create_pretrained(opt.weights, opt.weights)