Kai Zhang (github: https://github.com/cszn)
(img, mode=0)
| 378 | |
| 379 | |
| 380 | def augment_img(img, mode=0): |
| 381 | '''Kai Zhang (github: https://github.com/cszn) |
| 382 | ''' |
| 383 | if mode == 0: |
| 384 | return img |
| 385 | elif mode == 1: |
| 386 | return np.flipud(np.rot90(img)) |
| 387 | elif mode == 2: |
| 388 | return np.flipud(img) |
| 389 | elif mode == 3: |
| 390 | return np.rot90(img, k=3) |
| 391 | elif mode == 4: |
| 392 | return np.flipud(np.rot90(img, k=2)) |
| 393 | elif mode == 5: |
| 394 | return np.rot90(img) |
| 395 | elif mode == 6: |
| 396 | return np.rot90(img, k=2) |
| 397 | elif mode == 7: |
| 398 | return np.flipud(np.rot90(img, k=3)) |
| 399 | |
| 400 | |
| 401 | def augment_img_tensor4(img, mode=0): |
no outgoing calls
no test coverage detected