Commit 24981f66 authored by mac's avatar mac

no message

parent 009c5e1d
Pipeline #380 canceled with stages
......@@ -108,6 +108,7 @@ def img2Comix(args,net,device,image):
#print('----------torch.no_grad() handle')
#变成tensor数据后,形状变为CxHxW
image = to_tensor(image).unsqueeze(0) * 2 - 1
print('upsample_align=',args.upsample_align)
out = net(image.to(device), args.upsample_align).cpu()
out = out.squeeze(0).clip(-1, 1) * 0.5 + 0.5
# out = to_pil_image(out)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment