Spaces:
Runtime error
Runtime error
Commit ·
b4877a2
1
Parent(s): 19b1f35
Upload existing project files
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ transform = make_transform(size=(300, 450), normalize=True) # 图像变换
|
|
| 36 |
image_style = load_image('./data/udnie.jpg', transform=transform).to(device) # 风格图像
|
| 37 |
vgg = VGG(content_layers, style_layers).to(device) # 特征提取网络，只用来提取特征，不进行训练
|
| 38 |
model = TransNet(input_size=(300, 450)).to(device) # 内容生成网络，用于生成风格图片，进行训练
|
| 39 |
-
model.load_state_dict(torch.load('./models/udnie.pth'))
|
| 40 |
|
| 41 |
|
| 42 |
def process_images(image) :
|
|
|
|
| 36 |
image_style = load_image('./data/udnie.jpg', transform=transform).to(device) # 风格图像
|
| 37 |
vgg = VGG(content_layers, style_layers).to(device) # 特征提取网络，只用来提取特征，不进行训练
|
| 38 |
model = TransNet(input_size=(300, 450)).to(device) # 内容生成网络，用于生成风格图片，进行训练
|
| 39 |
+
model.load_state_dict(torch.load('./models/udnie.pth', map_location=device))
|
| 40 |
|
| 41 |
|
| 42 |
def process_images(image) :
|