Commit
·
227aa73
1
Parent(s):
e071b8d
updates
Browse files
detect.py
CHANGED
@@ -156,3 +156,8 @@ if __name__ == '__main__':
|
|
156 |
|
157 |
with torch.no_grad():
|
158 |
detect()
|
|
|
|
|
|
|
|
|
|
|
|
156 |
|
157 |
with torch.no_grad():
|
158 |
detect()
|
159 |
+
|
160 |
+
# Update all models
|
161 |
+
# for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
|
162 |
+
# detect()
|
163 |
+
# create_pretrained(opt.weights, opt.weights)
|
test.py
CHANGED
@@ -246,7 +246,7 @@ def test(data,
|
|
246 |
if __name__ == '__main__':
|
247 |
parser = argparse.ArgumentParser(prog='test.py')
|
248 |
parser.add_argument('--weights', type=str, default='weights/yolov5s.pt', help='model.pt path')
|
249 |
-
parser.add_argument('--data', type=str, default='data/
|
250 |
parser.add_argument('--batch-size', type=int, default=32, help='size of each image batch')
|
251 |
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
|
252 |
parser.add_argument('--conf-thres', type=float, default=0.001, help='object confidence threshold')
|
|
|
246 |
if __name__ == '__main__':
|
247 |
parser = argparse.ArgumentParser(prog='test.py')
|
248 |
parser.add_argument('--weights', type=str, default='weights/yolov5s.pt', help='model.pt path')
|
249 |
+
parser.add_argument('--data', type=str, default='data/coco128.yaml', help='*.data path')
|
250 |
parser.add_argument('--batch-size', type=int, default=32, help='size of each image batch')
|
251 |
parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
|
252 |
parser.add_argument('--conf-thres', type=float, default=0.001, help='object confidence threshold')
|