From 227aa735dccea28f44bc069b24a5a6356e9a707a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 24 Jun 2020 13:20:14 -0700 Subject: [PATCH] updates --- detect.py | 5 +++++ test.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/detect.py b/detect.py index e4a7ddb..2fb96ea 100644 --- a/detect.py +++ b/detect.py @@ -156,3 +156,8 @@ if __name__ == '__main__': with torch.no_grad(): 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) diff --git a/test.py b/test.py index ae6451f..04e1983 100644 --- a/test.py +++ b/test.py @@ -246,7 +246,7 @@ def test(data, if __name__ == '__main__': parser = argparse.ArgumentParser(prog='test.py') parser.add_argument('--weights', type=str, default='weights/yolov5s.pt', help='model.pt path') - parser.add_argument('--data', type=str, default='data/coco.yaml', help='*.data path') + parser.add_argument('--data', type=str, default='data/coco128.yaml', help='*.data path') parser.add_argument('--batch-size', type=int, default=32, help='size of each image batch') parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)') parser.add_argument('--conf-thres', type=float, default=0.001, help='object confidence threshold')