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

   

  
  ​
  ​