(h=256, w=256)
| 94 | |
| 95 | |
| 96 | def create_transforms_inference2(h=256, w=256): |
| 97 | transformer = transforms.Compose([ |
| 98 | transforms.Lambda(lambda img: F.rotate(img, angle=180)), |
| 99 | transforms.Resize(size=(h, w)), |
| 100 | transforms.ToTensor(), |
| 101 | ]) |
| 102 | |
| 103 | return transformer |
| 104 | |
| 105 | |
| 106 | def create_transforms_inference3(h=256, w=256): |