【GAN】pix2pix算法的数据集制作

最新推荐文章于 2024-10-15 12:40:18 发布

原创

最新推荐文章于 2024-10-15 12:40:18 发布 · 1.3k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#生成对抗网络 #算法 #人工智能

本文介绍了如何使用Python的argparse库处理A、B两个图像目录，生成imagepairs，并在PyCharm中设置图片路径。通过os和cv2库操作文件，实现图像的合并和保存。

一、A、B合并代码（此代码由官方提供）

import os
import numpy as np
import cv2
import argparse

parser = argparse.ArgumentParser('create image pairs')
parser.add_argument('--fold_A', dest='fold_A', help='input directory for image A', type=str, default='../dataset/50kshoes_edges')
parser.add_argument('--fold_B', dest='fold_B', help='input directory for image B', type=str, default='../dataset/50kshoes_jpg')
parser.add_argument('--fold_AB', dest='fold_AB', help='output directory', type=str, default