GAN网络进行图片增强

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

   

  
  ​
  ​