compare.py 文件源码

python
阅读 28 收藏 0 点赞 0 评论 0

项目:SketchSimplification 作者: La4La 项目源码 文件源码
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)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号