(image, alpha=0.2)
| 178 | |
| 179 | |
| 180 | def mixUp(image, alpha=0.2): |
| 181 | if alpha > 0: |
| 182 | lam = np.random.beta(alpha, alpha) |
| 183 | else: |
| 184 | lam = 1 |
| 185 | |
| 186 | batch_size = image.size()[0] |
| 187 | |
| 188 | assert batch_size > 2 |
| 189 | assert batch_size % 2 == 0 |
| 190 | |
| 191 | mixed_x = lam * image[0:int(batch_size / 2), |
| 192 | ...] + (1 - lam) * image[int(batch_size / 2):, ...] |
| 193 | |
| 194 | return mixed_x, lam |
| 195 | |
| 196 | |
| 197 | def mixUpCls(data, alpha=0.2): |