(image_name)
| 103 | |
| 104 | |
| 105 | def image_loader(image_name): |
| 106 | image = Image.open(image_name) |
| 107 | # fake batch dimension required to fit network's input dimensions |
| 108 | image = loader(image).unsqueeze(0) |
| 109 | return image.to(device, torch.float) |
| 110 | |
| 111 | |
| 112 | style_img = image_loader("./data/images/neural-style/picasso.jpg") |
no outgoing calls
no test coverage detected