File tree Expand file tree Collapse file tree 3 files changed +48
-3
lines changed
lib/src/otx/recipe/detection Expand file tree Collapse file tree 3 files changed +48
-3
lines changed Original file line number Diff line number Diff line change @@ -215,7 +215,22 @@ overrides:
215215 train_subset :
216216 batch_size : 8
217217 num_workers : 4
218- transforms : []
218+ transforms :
219+ - class_path : otx.data.transform_libs.torchvision.Resize
220+ init_args :
221+ scale : $(input_size)
222+ keep_ratio : false
223+ - class_path : otx.data.transform_libs.torchvision.RandomFlip
224+ init_args :
225+ probability : 0.5
226+ - class_path : torchvision.transforms.v2.ToDtype
227+ init_args :
228+ dtype : ${as_torch_dtype:torch.float32}
229+ scale : false
230+ - class_path : torchvision.transforms.v2.Normalize
231+ init_args :
232+ mean : [0.0, 0.0, 0.0]
233+ std : [255.0, 255.0, 255.0]
219234 sampler :
220235 class_path : otx.data.samplers.balanced_sampler.BalancedSampler
221236
Original file line number Diff line number Diff line change @@ -214,7 +214,22 @@ overrides:
214214 train_subset :
215215 batch_size : 8
216216 num_workers : 4
217- transforms : []
217+ transforms :
218+ - class_path : otx.data.transform_libs.torchvision.Resize
219+ init_args :
220+ scale : $(input_size)
221+ keep_ratio : false
222+ - class_path : otx.data.transform_libs.torchvision.RandomFlip
223+ init_args :
224+ probability : 0.5
225+ - class_path : torchvision.transforms.v2.ToDtype
226+ init_args :
227+ dtype : ${as_torch_dtype:torch.float32}
228+ scale : false
229+ - class_path : torchvision.transforms.v2.Normalize
230+ init_args :
231+ mean : [0.0, 0.0, 0.0]
232+ std : [255.0, 255.0, 255.0]
218233 sampler :
219234 class_path : otx.data.samplers.balanced_sampler.BalancedSampler
220235
Original file line number Diff line number Diff line change @@ -215,7 +215,22 @@ overrides:
215215 train_subset :
216216 batch_size : 8
217217 num_workers : 4
218- transforms : []
218+ transforms :
219+ - class_path : otx.data.transform_libs.torchvision.Resize
220+ init_args :
221+ scale : $(input_size)
222+ keep_ratio : false
223+ - class_path : otx.data.transform_libs.torchvision.RandomFlip
224+ init_args :
225+ probability : 0.5
226+ - class_path : torchvision.transforms.v2.ToDtype
227+ init_args :
228+ dtype : ${as_torch_dtype:torch.float32}
229+ scale : false
230+ - class_path : torchvision.transforms.v2.Normalize
231+ init_args :
232+ mean : [0.0, 0.0, 0.0]
233+ std : [255.0, 255.0, 255.0]
219234 sampler :
220235 class_path : otx.data.samplers.balanced_sampler.BalancedSampler
221236
You can’t perform that action at this time.
0 commit comments