diff --git a/yoltv4/eval.py b/yoltv4/eval.py index 682447a..81705e7 100644 --- a/yoltv4/eval.py +++ b/yoltv4/eval.py @@ -200,7 +200,7 @@ def precision_calc(proposal_polygons_dir, gt_polygons_dir, auto-created using all classes that appear ground truth polygons. threshold : float A value between 0.0 and 1.0 that determines the IOU threshold for a - true positve. + true positive. file_format : str The extension or file format for predictions Returns @@ -633,4 +633,4 @@ def mAP_score(proposal_polygons_dir, gt_polygons_dir, APs_by_class.append(AP) mAP = np.average(APs_by_class) print("mAP:", mAP, "@IOU:", threshold) - return object_subset, mAP, APs_by_class, mF1_score, f1s_by_class, precision_iou_by_obj, precision_by_class, mPrecision, recall_iou_by_obj, recall_by_class, mRecall, confidences \ No newline at end of file + return object_subset, mAP, APs_by_class, mF1_score, f1s_by_class, precision_iou_by_obj, precision_by_class, mPrecision, recall_iou_by_obj, recall_by_class, mRecall, confidences