一、训练图片的制作
该部分主要参考博文https://blog.csdn.net/qq_27871973/article/details/85009026
1.将图片转化为固定大小
我建立了一个文件夹专门用来前期处理图片使用
I第一个文件夹是处理图片的代码,
在这里插入代码片
###将oldimage文件夹中图片批量重新裁剪至416*416大小按顺序保存至newimage中###
import cv2
import os
image_size=416
source_path="F:/picture_processing/oldimage/"
target_path="F:/picture_processing/newimage/"
if not os.path.exists(target_path):
os.makedirs(target_path)
def rename(path):
# 给图片编队函数
filelist = os.listdir(path) # 获取文件路径
total_num = len(filelist) # 获取文件长度
i = 1 # 文件从1开始命名
for item in filelist:
if item.endswith('.jpg'):
src = os.path.join(os.path.abspath(path), item)
# dst = os.path.join(os.path.abspath(path), ''+str(i) + '.jpg')
dst = os.path.join(os.path.abspath(path), '00' + format(str(i), '0>3s') + '.jpg')
try:
os.rename(src, dst)
print('converting %s to %s ...' % (src, dst))
i = i + 1
except:
continue
print('total %d to rename & converted %d jpgs' % (total_num, i))
if __name__=="__main__":
image_list=os.listdir(source_path)
i=0
for image in image_list:
i=i+1
image_source=cv2.imread(source_path+image)
print("正在处理第"+str(i)+"张图片")
image=cv2.resize(image_source,(image_size,image_size),interpolation=cv2.INTER_CUBIC)
cv2.imwrite(target_path+str(i)+".jpg",image)
rename(target_path)
print("处理完成!")
然后使用 labelImg软件,对其进行标注。具体做法仍是参考该博文