ultralytics 8.0.151 add DOTAv2.yaml for OBB training (#4258)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
This commit is contained in:
parent
a76af55533
commit
c9be1f3cce
46 changed files with 805 additions and 303 deletions
|
|
@ -246,7 +246,7 @@ fold_lbl_distrb.to_csv(save_path / "kfold_label_distribution.csv")
|
|||
results = {}
|
||||
for k in range(ksplit):
|
||||
dataset_yaml = ds_yamls[k]
|
||||
model.train(data=dataset_yaml, *args, **kwargs) # Include any training arguments
|
||||
results = model.train(data=dataset_yaml, *args, **kwargs) # Include any training arguments
|
||||
results[k] = model.metrics # save output metrics for further analysis
|
||||
```
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue