GAN网络进行图片增强

基于Tensorflow框架

调用ModifyPictureSize.py文件

代码如下：

from skimage import io,transform,color

import numpy as np



def convert_gray(f,**args):

    """

    将彩色图片转换为灰度图片和调整大小，改变图像分辨率

    :return:

    """

    rgb = io.imread(f)

    gray = color.rgb2gray(rgb)  #

    dst = transform.resize(gray, (28, 28))

    return dst



datapath='yourpath'

str=datapath+'/*.jpg'   #识别.jpg的图像

coll = io.ImageCollection(str,load_func=convert_gray)#批处理

for i in range(len(coll)):

    io.imsave(r'yourpath'+np.str(i)+'.png',coll[i])

将图片转换为二进制文件

1.将你的jpg和/或png图像复制到一个类文件夹中

2.更改相应的标签 batches.meta.txt

3.然后使用以下bash脚本处理图像，重新调整你放在文件夹中的所有png MNIST标准28x28pixel大小
```
source ./resize-script.sh
```
4.最后运行以下python脚本将所有图片和类别折叠成单个ble二进制文件 - 二进制文件将显示为ubyte准备好tar的文件
```
python convert-images-to-mnist-format.py
```

GAN网络模型进行训练

python main.py --dataset mnist --gan_type GAN --epoch 25 --batch_size 64

巴特西

GAN网络进行图片增强

GAN网络进行图片增强

基于Tensorflow框架

调用ModifyPictureSize.py文件

将图片转换为二进制文件

GAN网络模型进行训练

最新文章

热门文章