Rename metric.py to metrics.py

furyhawk · Sep 7, 2022 · f06b323 · f06b323
1 parent 3b8a3fb
commit f06b323
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolov6/utils/metric.py → yolov6/utils/metrics.py b/yolov6/utils/metric.py → yolov6/utils/metrics.py
@@ -164,4 +164,4 @@ def process_batch(detections, labels, iouv):
                 # matches = matches[matches[:, 2].argsort()[::-1]]
                 matches = matches[np.unique(matches[:, 0], return_index=True)[1]]
             correct[matches[:, 1].astype(int), i] = True
-    return torch.tensor(correct, dtype=torch.bool, device=iouv.device)
+    return torch.tensor(correct, dtype=torch.bool, device=iouv.device)