diff --git a/detect.py b/detect.py index 924fb48..38de3aa 100644 --- a/detect.py +++ b/detect.py @@ -230,8 +230,8 @@ def parse_opt(): # parser.add_argument('--source', type=str, default=ROOT / 'data/images', help='file/dir/URL/glob, 0 for webcam') # parser.add_argument('--data', type=str, default=ROOT / 'data/coco128.yaml', help='(optional) dataset.yaml path') - parser.add_argument('--weights', nargs='+', type=str, default=ROOT / 'weights/best.pt', help='model path(s)') - parser.add_argument('--source', type=str, default=ROOT / 'D:\\dataSet\\VOCdevkit\\images\\val', help='file/dir/URL/glob, 0 for webcam') + parser.add_argument('--weights', nargs='+', type=str, default=ROOT / 'runs/train/exp5/weights/best.pt', help='model path(s)') + parser.add_argument('--source', type=str, default=ROOT / '../VOCdevkit/images/val', help='file/dir/URL/glob, 0 for webcam') parser.add_argument('--batch-size', type=int, default=4, help='total batch size for all GPUs, -1 for autobatch') parser.add_argument('--data', type=str, default=ROOT / 'data/forest.yaml', help='(optional) dataset.yaml path') parser.add_argument('--workers', type=int, default=0, help='max dataloader workers (per RANK in DDP mode)')