Args: dataset(Dataset) : Pytorch dataset object. img_size (tuple): mosaic (bool): enable mosaic augmentation or not. preproc (func): degrees (float): translate (float): scale (tuple): mscale (tu
(
self, dataset, img_size, mosaic=True, preproc=None,
degrees=10.0, translate=0.1, scale=(0.5, 1.5), mscale=(0.5, 1.5),
shear=2.0, perspective=0.0, enable_mixup=True, *args
)
| 38 | """Detection dataset wrapper that performs mixup for normal dataset.""" |
| 39 | |
| 40 | def __init__( |
| 41 | self, dataset, img_size, mosaic=True, preproc=None, |
| 42 | degrees=10.0, translate=0.1, scale=(0.5, 1.5), mscale=(0.5, 1.5), |
| 43 | shear=2.0, perspective=0.0, enable_mixup=True, *args |
| 44 | ): |
| 45 | """ |
| 46 | |
| 47 | Args: |
| 48 | dataset(Dataset) : Pytorch dataset object. |
| 49 | img_size (tuple): |
| 50 | mosaic (bool): enable mosaic augmentation or not. |
| 51 | preproc (func): |
| 52 | degrees (float): |
| 53 | translate (float): |
| 54 | scale (tuple): |
| 55 | mscale (tuple): |
| 56 | shear (float): |
| 57 | perspective (float): |
| 58 | enable_mixup (bool): |
| 59 | *args(tuple) : Additional arguments for mixup random sampler. |
| 60 | """ |
| 61 | super().__init__(img_size, mosaic=mosaic) |
| 62 | self._dataset = dataset |
| 63 | self.preproc = preproc |
| 64 | self.degrees = degrees |
| 65 | self.translate = translate |
| 66 | self.scale = scale |
| 67 | self.shear = shear |
| 68 | self.perspective = perspective |
| 69 | self.mixup_scale = mscale |
| 70 | self.enable_mosaic = mosaic |
| 71 | self.enable_mixup = enable_mixup |
| 72 | |
| 73 | def __len__(self): |
| 74 | return len(self._dataset) |
nothing calls this directly
no outgoing calls
no test coverage detected