(h=256, w=256)
| 104 | |
| 105 | |
| 106 | def create_transforms_inference3(h=256, w=256): |
| 107 | transformer = transforms.Compose([ |
| 108 | transforms.Lambda(lambda img: F.rotate(img, angle=270)), |
| 109 | transforms.Resize(size=(h, w)), |
| 110 | transforms.ToTensor(), |
| 111 | ]) |
| 112 | |
| 113 | return transformer |
| 114 | |
| 115 | |
| 116 | def create_transforms_inference4(h=256, w=256): |