diff --git a/test.py b/test.py index 350a6a1f..9c0bae55 100644 --- a/test.py +++ b/test.py @@ -174,7 +174,7 @@ if __name__ == '__main__': parser.add_argument('--iou-thres', type=float, default=0.5, help='iou threshold required to qualify as detected') parser.add_argument('--conf-thres', type=float, default=0.3, help='object confidence threshold') parser.add_argument('--nms-thres', type=float, default=0.45, help='iou threshold for non-maximum suppression') - parser.add_argument('--coco_map', action='store_true', help='use pycocotools mAP') + parser.add_argument('--coco-map', action='store_true', help='use pycocotools mAP') parser.add_argument('--img-size', type=int, default=416, help='size of each image dimension') opt = parser.parse_args() print(opt, end='\n\n') diff --git a/utils/gcp.sh b/utils/gcp.sh index 342653c1..7315c3c2 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -7,7 +7,7 @@ bash yolov3/data/get_coco_dataset.sh # Start sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 sudo rm -rf cocoapi && git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make && cd ../.. && cp -r cocoapi/PythonAPI/pycocotools yolov3 -cd yolov3 && python3 train.py +cd yolov3 && python3 test.py --coco-map # Resume python3 train.py --resume