(self, data: torch.Tensor, labels: torch.Tensor | None = None, randomize=True)
| 84 | return mixweight * data + (1 - mixweight) * data[perm, ...] |
| 85 | |
| 86 | def __call__(self, data: torch.Tensor, labels: torch.Tensor | None = None, randomize=True): |
| 87 | data_t = convert_to_tensor(data, track_meta=get_track_meta()) |
| 88 | labels_t = data_t # will not stay this value, needed to satisfy pylint/mypy |
| 89 | if labels is not None: |
| 90 | labels_t = convert_to_tensor(labels, track_meta=get_track_meta()) |
| 91 | if randomize: |
| 92 | self.randomize() |
| 93 | if labels is None: |
| 94 | return convert_to_dst_type(self.apply(data_t), dst=data)[0] |
| 95 | |
| 96 | return ( |
| 97 | convert_to_dst_type(self.apply(data_t), dst=data)[0], |
| 98 | convert_to_dst_type(self.apply(labels_t), dst=labels)[0], |
| 99 | ) |
| 100 | |
| 101 | |
| 102 | class CutMix(Mixer): |
nothing calls this directly
no test coverage detected