Add bgr hyperparameter (#9139)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
e706c7f625
commit
33607ca278
8 changed files with 13 additions and 2 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue