tiny-yoloV3消化之旅（一）

一、训练图片的制作
该部分主要参考博文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软件，对其进行标注。具体做法仍是参考该博文

tiny-yoloV3消化之旅（一）

猜你喜欢