31 lines
946 B
YAML
31 lines
946 B
YAML
|
|
train_dataloader:
|
|
dataset:
|
|
return_masks: False
|
|
transforms:
|
|
ops:
|
|
- {type: RandomPhotometricDistort, p: 0.5}
|
|
- {type: RandomZoomOut, fill: 0}
|
|
- {type: RandomIoUCrop, p: 0.8}
|
|
- {type: SanitizeBoundingBoxes, min_size: 1}
|
|
- {type: RandomHorizontalFlip}
|
|
- {type: Resize, size: [640, 640], }
|
|
- {type: SanitizeBoundingBoxes, min_size: 1}
|
|
- {type: ConvertPILImage, dtype: 'float32', scale: True}
|
|
- {type: ConvertBoxes, fmt: 'cxcywh', normalize: True}
|
|
collate_fn:
|
|
type: BatchImageCollateFunction
|
|
scales: [480, 512, 544, 576, 608, 640, 640, 640, 672, 704, 736, 768, 800]
|
|
shuffle: True
|
|
num_workers: 4
|
|
total_batch_size: 16
|
|
|
|
val_dataloader:
|
|
dataset:
|
|
transforms:
|
|
ops:
|
|
- {type: Resize, size: [640, 640]}
|
|
- {type: ConvertPILImage, dtype: 'float32', scale: True}
|
|
shuffle: False
|
|
total_batch_size: 16
|
|
num_workers: 8 |