目标检测——自定义DataSet类

介绍写一个属于自己的DataSet，通过这个数据集我们就可以去读取PASCAL VOC的数据，或者读取自己数据集的数据。

1.根据数据集如何生成train.txt 和val.txt

其实这个比较简单，通过运行脚本split_data.py可以生成train.txt和val.txt

import os
import random


def main():
    random.seed(0)  # 设置随机种子，保证随机结果可复现
	# .xml格式的标注文件的根目录或者保存的.jpg图片的根目录
    files_path = "./VOCdevkit/VOC2012/Annotations"
版权声明：本文为weixin_38346042原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。