#coding:utf-8
from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
from torchvision import transforms
import GeneralTransfer,SuperTransfer,QinghuaTransfer,Display
from QinghuaTransfer import im_convert,load_image


if __name__ == '__main__':
    #图片地址
    content_dir = "data/Test/test4/content4.jpg"
    style_dir = "data/Test/test4/style4.jpg"
    result_dir = "data/Test/test4/result"
    result5000_dir = result_dir + "10000.bmp"
    #运行青花风格迁移
    QinghuaTransfer.main(content_dir,style_dir,result_dir)
    #SuperTransfer.main(content_dir,style_dir,result_dir)
    '''
    content = load_image(content_dir, max_size=400)
    style = load_image(style_dir, shape=content.shape[-2:])
    result = load_image(result5000_dir, max_size=400)
    fig, (ax1, ax2, ax3) = plt.subplots(1, 3, figsize=(12, 5))
    ax1.imshow(im_convert(content))
    ax1.set_title("content")
    ax2.imshow(im_convert(style))
    ax2.set_title("style")
    ax3.imshow(im_convert(result))
    ax3.set_title("result")
    plt.show()
    '''