add export to FP16

add inference with possibility to have different
This commit is contained in:
zxsanny
2025-03-28 12:54:25 +02:00
parent eaef1a9b66
commit 5b89a21b36
9 changed files with 365 additions and 242 deletions
+1
View File
@@ -184,6 +184,7 @@ def train_dataset(existing_date=None, from_scratch=False):
model_name = latest_model if latest_model is not None and path.isfile(latest_model) and not from_scratch else 'yolo11m.yaml'
print(f'Initial model: {model_name}')
model = YOLO(model_name)
model.info['author'] = 'LLC Azaion'
yaml = abspath(path.join(cur_dataset, 'data.yaml'))
results = model.train(data=yaml,