背景:
毕设设计相关,通过谷歌的tensorflow训练一个关于深度学习的图像识别,但是苦于原始数据集太过稀疏(就是很少),故通过数据增强的方法来对原始数据进行水平翻转等操作,这里谢谢学长为我们踩过的坑。
也很欣慰昨天终于能跑出来一个模型了,这之前图的标记却已进行近一个月....
要求:
该代码的运行 root 变量指定的是文件夹,存储文件夹是 storage 来记录,同时有两个约数变量 save_dirs 和 begin_rename_num ,前者指定是否存储到一个单独的文件夹,默认为 1,后者是对处理后的文件进行重命名指定开始的数字。。。。。啊,我的天,编不下去了,看代码看代码。。
学长要求:
原图命名为六位数字的png文件,标记图命名为六位数字+_L的 png 文件
参考代码:
2019年6月28日15:36:02 优化代码流程
#!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Time :2019/04/16 09:21
# @Author : Qdan
# @File :新项目.py
# @Project :PyCharm
# @Description: 对图片数据集进行操作
# 项目要求:对给定目录下的三个子文件夹进行处理并保存在增强数据集子文件夹下
#
import os
from PIL import Image
# 工作路径
working_path = r"D:\Documents\Workspaces\毕业设计项目\00全局项目\堇青石角石"
# runing
##############################################################################################################
folders = os.listdir(working_path)
def pro_pictures(image_path, suffix, new_dir):
"""
对给定图片进行数据增强处理
:param image_path: 图片所在路径
:param suffix: 序列图片后缀填充
:param new_dir: 存储的新路径
:return: 空值
"""
rename_num = 1 # 命名的起始编号
parent = image_path # 图片路径
fill_rename_num = 6 # 填充的位数
suffix = suffix # 标志填充字符
filenames = os.listdir(image_path)
save_image = working_path + r"\\处理数据集\\"
save_image = save_image + new_dir
if not os.path.exists(save_image):
os.makedirs(save_image)
for filename in filenames:
# 获取当前路径下的当前文件
current_path = os