xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix non-coco val

Browse Source
This commit is contained in:
wa22 2024-05-24 07:38:21 +00:00
parent 483d7a9050
commit 66bca50c47
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ultralytics/engine/validator.py
View File

@ -112,7 +112,7 @@ class BaseValidator:
if self.training: if self.training:
self.device = trainer.device self.device = trainer.device
self.data = trainer.data self.data = trainer.data
self.args.half = self.device.type != "cpu" # force FP16 val during training # self.args.half = self.device.type != "cpu" # force FP16 val during training
model = trainer.ema.ema or trainer.model model = trainer.ema.ema or trainer.model
model = model.half() if self.args.half else model.float() model = model.half() if self.args.half else model.float()
# self.model = model # self.model = model
@ -196,11 +196,12 @@ class BaseValidator:
self.check_stats(stats) self.check_stats(stats)
self.speed = dict(zip(self.speed.keys(), (x.t / len(self.dataloader.dataset) * 1e3 for x in dt))) self.speed = dict(zip(self.speed.keys(), (x.t / len(self.dataloader.dataset) * 1e3 for x in dt)))
self.finalize_metrics() self.finalize_metrics()
# self.print_results() if not (self.args.save_json and self.jdict):
self.print_results()
self.run_callbacks("on_val_end") self.run_callbacks("on_val_end")
if self.training: if self.training:
model.float() model.float()
assert(self.args.save_json and self.jdict) if self.args.save_json and self.jdict:
with open(str(self.save_dir / "predictions.json"), "w") as f: with open(str(self.save_dir / "predictions.json"), "w") as f:
LOGGER.info(f"Saving {f.name}...") LOGGER.info(f"Saving {f.name}...")
json.dump(self.jdict, f) # flatten and save json.dump(self.jdict, f) # flatten and save