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

Allocated updated pycocotools metrics fix (#101)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Laughing 2022-12-27 22:49:13 +08:00 • committed by GitHub
parent 441126a77e
commit 5c6d11bdb2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/yolo/engine/validator.py
View file

@ -127,7 +127,7 @@ class BaseValidator:
self.logger.info(f"Saving {f.name}...")
json.dump(self.jdict, f) # flatten and save
self.eval_json()
stats = self.eval_json(stats)
return stats
def get_dataloader(self, dataset_path, batch_size):
@ -171,5 +171,5 @@ class BaseValidator:
def pred_to_json(self, preds, batch):
pass
def eval_json(self):
def eval_json(self, stats):
pass