(h=256, w=256)
| 114 | |
| 115 | |
| 116 | def create_transforms_inference4(h=256, w=256): |
| 117 | transformer = transforms.Compose([ |
| 118 | transforms.Lambda(lambda img: F.hflip(img)), |
| 119 | transforms.Resize(size=(h, w)), |
| 120 | transforms.ToTensor(), |
| 121 | ]) |
| 122 | |
| 123 | return transformer |
| 124 | |
| 125 | |
| 126 | def create_transforms_inference5(h=256, w=256): |