From 41ba6dfd6b80fc6550f20da1d8661fa46870d343 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 26 Feb 2019 03:12:04 +0100 Subject: [PATCH] updates --- test.py | 4 +++- utils/gcp.sh | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/test.py b/test.py index 8d4f8646..350a6a1f 100644 --- a/test.py +++ b/test.py @@ -174,6 +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('--img-size', type=int, default=416, help='size of each image dimension') opt = parser.parse_args() print(opt, end='\n\n') @@ -187,7 +188,8 @@ if __name__ == '__main__': opt.img_size, opt.iou_thres, opt.conf_thres, - opt.nms_thres + opt.nms_thres, + opt.coco_map ) # Image Total P R mAP # YOLOv3 320 diff --git a/utils/gcp.sh b/utils/gcp.sh index 54bc8ae9..342653c1 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -5,7 +5,9 @@ sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 bash yolov3/data/get_coco_dataset.sh # Start -sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py +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 # Resume python3 train.py --resume