Commit dbba7b70 authored by mac's avatar mac

c

parent dc7edf7c
Pipeline #369 canceled with stages
......@@ -99,7 +99,7 @@ def img2Comix(args,net,device,image):
#image = load_image(os.path.join(args.input_dir, image_name), args.x32)
#image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image = img_2_x32(image,args.x32)
with torch.no_grad():
......@@ -119,7 +119,7 @@ def img2Comix(args,net,device,image):
out = 255 * out
print(type(out))
print(out.shape)
print(out)
#print(out)
return 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