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