Add bgr hyperparameter (#9139)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Laughing 2024-03-24 06:17:00 +08:00 committed by GitHub
parent e706c7f625
commit 33607ca278
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 13 additions and 2 deletions

View file

@ -167,6 +167,7 @@ class YOLODataset(BaseDataset):
batch_idx=True,
mask_ratio=hyp.mask_ratio,
mask_overlap=hyp.overlap_mask,
bgr=hyp.bgr if self.augment else 0.0, # only affect training.
)
)
return transforms