Update metrics names (#85)

This commit is contained in:
Glenn Jocher 2022-12-24 02:32:24 +01:00 committed by GitHub
parent 6432afc5f9
commit 248d54ca03
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 30 additions and 36 deletions

View file

@ -178,12 +178,12 @@ class SegmentationValidator(DetectionValidator):
return [
"metrics/precision(B)",
"metrics/recall(B)",
"metrics/mAP_0.5(B)",
"metrics/mAP_0.5:0.95(B)", # metrics
"metrics/mAP50(B)",
"metrics/mAP50-95(B)", # metrics
"metrics/precision(M)",
"metrics/recall(M)",
"metrics/mAP_0.5(M)",
"metrics/mAP_0.5:0.95(M)",]
"metrics/mAP50(M)",
"metrics/mAP50-95(M)",]
def plot_val_samples(self, batch, ni):
images = batch["img"]