From c1ab7237767d7dec6533df5ea9f2144dbe319d04 Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephane.chavin@lis-lab.fr>
Date: Thu, 12 Dec 2024 11:10:13 +0100
Subject: [PATCH] save TP and FP for validation
---
yolov5/val.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/yolov5/val.py b/yolov5/val.py
index d4073b4..88e7454 100755
--- a/yolov5/val.py
+++ b/yolov5/val.py
@@ -274,6 +274,8 @@ def run(
stats = [torch.cat(x, 0).cpu().numpy() for x in zip(*stats)] # to numpy
if len(stats) and stats[0].any():
tp, fp, p, r, f1, ap, ap_class = ap_per_class(*stats, plot=plots, save_dir=save_dir, names=names)
+ np.save(save_dir / 'class_tp.npy', tp)
+ np.save(save_dir / 'class_fp.npy', fp)
ap50, ap = ap[:, 0], ap.mean(1) # AP@0.5, AP@0.5:0.95
mp, mr, map50, map = p.mean(), r.mean(), ap50.mean(), ap.mean()
nt = np.bincount(stats[3].astype(int), minlength=nc) # number of targets per class
--
GitLab