updates

2018-09-10 16:41:02 +02:00
parent ff04315f96
commit 300e2b5dfc
3 changed files with 10 additions and 2 deletions
@@ -12,3 +12,11 @@ python3 detect.py

 # Test
 python3 test.py -img_size 416 -weights_path checkpoints/yolov3.weights
+
+
+sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3
+cd yolov3
+cd checkpoints
+wget https://pjreddie.com/media/files/yolov3.weights
+cd ..
+python3 test.py -img_size 416 -weights_path checkpoints/yolov3.weights
@@ -99,7 +99,7 @@ def ap_per_class(tp, conf, pred_cls, target_cls):
    tp, conf, pred_cls = tp[i], conf[i], pred_cls[i]

    # Find unique classes
-    unique_classes = target_cls# np.unique(np.concatenate((pred_cls, target_cls), 0))
+    unique_classes = np.unique(np.concatenate((pred_cls, target_cls), 0))

    # Create Precision-Recall curve and compute AP for each class
    ap = []