1. 关于数据集预处理
把训练数据train和测试数据test的img和gt,放到datasets文件夹下,或者其他文件夹中也可以,代码中写出数据集所在位置的绝对路径即可。
将训练数据和测试数据生成如下图的格式:
参考文献
生成train.txt和test.txt,保存到datasets文件夹下(必须也把test.txt也生成)。
生成文件的代码如下:
import os
def get_images(img_path):
'''
find image files in data path
:return: list of files found
'''
img_path = os.path.abspath(img_path)
files = []
exts = ['jpg', 'png', 'jpeg', 'JPG', 'PNG']
for parent