1 将要训练的图片路径读取成txt格式
2将xml-txt 格式
代码如下 容易理解
参考博客 https://blog.csdn.net/u012966194/article/details/80004647
感谢提供参考与支持
path = "images/"
for filenames in os.walk(pathh):
filenames = list(filenames)
filenames = filenames[2]
for filename in filenames:
print(filename)
with open ("class_train1.txt",'a') as f:
f.write(path+filename+'\n')
# -*- coding: utf-8 -*-
import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join
sets = []
classes = ["dog","person","cat"]
#原样保留。size为图片大小
# 将ROI的坐标转换为yolo需要的坐标
# size是图片的w和h
# box里保存的是ROI的坐标(x,y的最大值和最小值)
# 返回值为ROI中心点相对于图片大小的比例坐标,和ROI的w、h相对于图片大小的比例
def convert(size, box):
dw = 1./(size[0])
dh =