def save_as_img(model1, model2, name, origin, transposed=False):
if transposed:
origin = origin.transpose(2, 1, 0)
model1 = model1.transpose(2, 1, 0)
model2 = model2.transpose(2, 1, 0)
else:
origin = origin.transpose(1, 2, 0)
model1 = model1.transpose(1, 2, 0)
model2 = model2.transpose(1, 2, 0)
model1 = model1 * 255
model1 = model1.clip(0, 255).astype(np.uint8)
img1 = cuda.to_cpu(model1)
model2 = model2 * 255
model2 = model2.clip(0, 255).astype(np.uint8)
img2 = cuda.to_cpu(model2)
origin = origin.clip(0, 255).astype(np.uint8)
img_concat = cv2.hconcat([origin, img1])
img_concat = cv2.hconcat([img_concat, img2])
cv2.imwrite(name, img_concat)
评论列表
文章目录