HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Update metrics names (#85)

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/v8/detect/val.py
View file

@ -173,7 +173,7 @@ class DetectionValidator(BaseValidator):
# TODO: align with train loss metrics
@property
def metric_keys(self):
return ["metrics/precision(B)", "metrics/recall(B)", "metrics/mAP_0.5(B)", "metrics/mAP_0.5:0.95(B)"]
return ["metrics/precision(B)", "metrics/recall(B)", "metrics/mAP50(B)", "metrics/mAP50-95(B)"]
def plot_val_samples(self, batch, ni):
images = batch["img"]