如何使用stylegan训练自己的照片数据集

今天完成了简单的stylegan自定义数据集训练过程,感谢国外老哥的教程,原来很简单……其实对于专门做人工智能方面的人来说训练应该是很简单的,但是对于我这种跨专业跨领域的小白来说,还是需要一个详细教程的,这里给大家介绍一下如何使用自己的数据集来测试stylegan。 首先,大家需要在自己的电脑上配置好tensorflow环境,注意:tensorflow2.0可能无法运行stylegan的原始代码,建议使用1.13版本。 个人运行环境:WIN10+CUDA10.0 +CUDNN7.5+ TENSORFLOW-GPU1.13.1 + python3.7

使用python对文件夹内的照片进行批量修改尺寸

小白学习记录……

原文链接:https://blog.csdn.net/atyzy/article/details/77905463

使用方法……这个很简单, 新建resize.py文件,复制上面代码,修改代码中这三个路径:

filename = os.listdir(“C:/Users/Administrator/stylegan/swift/”)    (这个就是需要修改的图片所在文件夹)

base_dir = “C:/Users/Administrator/stylegan/swift/”   (我也不晓得啥意思,反正上面一样就行了)

new_dir = “C:/Users/Administrator/stylegan/swift2/”   (新文件输出的路径位置)

这里要注意一点,文件路径要改为/斜杠而不是win路径的\,否则会爆出(unicode error) ‘unicodeescape’ codec can’t decode bytes in position 2-3: tr这个错误

或者你可以使用\\双斜杠来代替………………(大概因为我是小白才搞出这个错误吧……)

size_m = 128

size_n = 128

这个是尺寸,宽高像素……

然后照样在终端或者anaconda prompt中输入 python resize.py就可以运行了

使用python对文件夹内的照片进行批量重命名

原代码来自:https://www.jb51.net/article/156684.htm

找来找去终于看到一个能用的,简单修改好以后可以使用python3或者anaconda prompt进行批量文件名修改了……

使用方法很简单,

1.先使用随便一个代码编辑器,比如vs code,新建一个rename.py文件

2.将上面代码复制进去,修改 self.path = C:/Users/ThinkPad User/Desktop/weibo 这里的路径信息为你需要修改的文件夹信息……

3.保存

4.打开终端或者你的anaconda prompt 输入cd C:/Users/ThinkPad User/Desktop/weibo (这里是打开你的rename.py所在文件夹的路径)

5.终端或者你的anaconda prompt 中输入python rename.py 

6.等待批量修改完成