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

Detection training zero-mAP fix (#192)

Browse Source
This commit is contained in:
Glenn Jocher 2023-01-10 12:18:39 +01:00 committed by GitHub
parent a8a70390ab
commit 0a7d86cd26
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/yolo/engine/exporter.py
View File

@ -168,7 +168,7 @@ class Exporter:
# Input
im = torch.zeros(self.args.batch, 3, *self.imgsz).to(self.device)
file = Path(getattr(model, 'pt_path', None) or model.yaml['yaml_file'])
file = Path(getattr(model, 'pt_path', None) or getattr(model, 'yaml_file', None) or model.yaml['yaml_file'])
if file.suffix == '.yaml':
file = Path(file.name)

1
ultralytics/yolo/engine/model.py
View File

@ -218,3 +218,4 @@ class YOLO:
args.pop("batch", None)
args.pop("epochs", None)
args.pop("cache", None)
args.pop("save_json", None)

2
ultralytics/yolo/v8/detect/train.py
View File

@ -60,7 +60,7 @@ class DetectionTrainer(BaseTrainer):
def get_model(self, cfg=None, weights=None, verbose=True):
model = DetectionModel(cfg, ch=3, nc=self.data["nc"], verbose=verbose)
if weights:
model.load(model)
model.load(weights)
return model