zoukankan      html  css  js  c++  java
  • 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


       

  • 相关阅读:
    centos7源以及相关的一些命令
    创建Vue实例的三种方法
    github 钩子管理工具 overcommit
    npm管理registry 【转】
    两个字典增量部分
    celery (二) task调用
    shell编程
    linux 环境变量 转
    celery (二) task
    celery(一) application
  • 原文地址:https://www.cnblogs.com/shierlou-123/p/11283814.html
Copyright © 2011-2022 走看看