updates

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

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