xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix load and resume and update autodownload endpoint (#136)

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:
Ayush Chaurasia
2023-01-03 01:42:01 +05:30
committed by GitHub
parent 6d5123297e
commit 82c849c163
5 changed files with 16 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -54,12 +54,9 @@ class DetectionTrainer(BaseTrainer):
self.model.names = self.data["names"]
def load_model(self, model_cfg=None, weights=None, verbose=True):
model = DetectionModel(model_cfg or getattr(weights, 'yaml', None) or weights['model'].yaml,
ch=3,
nc=self.data["nc"],
verbose=verbose)
model = DetectionModel(model_cfg or weights.yaml, ch=3, nc=self.data["nc"], verbose=verbose)
if weights:
model.load(weights['model'] if isinstance(weights, dict) else weights, verbose)
model.load(weights, verbose)
return model
def get_validator(self):

7
ultralytics/yolo/v8/segment/train.py
View File

@ -17,12 +17,9 @@ from ultralytics.yolo.utils.torch_utils import de_parallel
class SegmentationTrainer(v8.detect.DetectionTrainer):
def load_model(self, model_cfg=None, weights=None, verbose=True):
model = SegmentationModel(model_cfg or getattr(weights, 'yaml', None) or weights['model'].yaml,
ch=3,
nc=self.data["nc"],
verbose=verbose)
model = SegmentationModel(model_cfg or weights.yaml, ch=3, nc=self.data["nc"], verbose=verbose)
if weights:
model.load(weights['model'] if isinstance(weights, dict) else weights, verbose)
model.load(weights, verbose)
return model
def get_validator(self):