Spaces:
Running
Running
| import cv2,os | |
| import AnimeGANv3_src | |
| if __name__ == '__main__': | |
| f = 'A' | |
| input_imgs_path = r'../../v3-usa\dataset\USA\val' | |
| # input_imgs_path = r'/mnt/data/xinchen/v3-usa/dataset/USA/val' | |
| output_path = 'AnimeGANv3_usa_64_output' | |
| # img = cv2.imread(os.path.join(input_imgs_path, os.listdir(input_imgs_path)[0])) | |
| img = cv2.imread(os.path.join(input_imgs_path, 'jp_16.png')) | |
| out = AnimeGANv3_src.Convert(cv2.cvtColor(img, cv2.COLOR_BGR2RGB), f, True) | |
| # cv2.imshow('d', cv2.cvtColor(out, cv2.COLOR_BGR2RGB)) | |
| # cv2.waitKey(0) | |
| cv2.imwrite('a.jpg', cv2.cvtColor(out, cv2.COLOR_BGR2RGB)) | |