Refactoring cleanup (#3565)

* Refactoring cleanup

* Update test.py
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-06-09 22:50:27 +02:00
1 parent 0e5cfdbea7
commit 4695ca8314
3 files changed
+29 -27

No files matched your search

+1 -1
View File
@@ -178,7 +178,7 @@ if __name__ == '__main__':
parser.add_argument('--source', type=str, default='data/images', help='file/dir/URL/glob, 0 for webcam')
parser.add_argument('--imgsz', '--img', '--img-size', type=int, default=640, help='inference size (pixels)')
parser.add_argument('--conf-thres', type=float, default=0.25, help='confidence threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='NMS IOU threshold')
parser.add_argument('--iou-thres', type=float, default=0.45, help='NMS IoU threshold')
parser.add_argument('--max-det', type=int, default=1000, help='maximum detections per image')
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
parser.add_argument('--view-img', action='store_true', help='show results')