目标检测算法丨YOLOv3数据处理
2022-10-17 11:58 作者:腾科IT教育官方账号 | 我要投稿
数据集
VOC2007数据集有20个类:aeroplane, bicycle, bird, boat, bottle, bus, car, cat, chair, cow, diningtable, dog, horse, motorbike, person, pottedplant, sheep, sofa, train, tv/monitor。
VOC数据集下载地址如下,里面已经包括了训练集、测试集、验证集(与测试集一样),无需再次划分:
链接: https://pan.baidu.com/s/19Mw2u_df_nBzsC2lg20fQA
提取码: j5ge
解压出来之后,下面的目录结构

生成数据
拿到数据之后,我们还需要利用上面的信息,生成模型需要的形式。
首先我们需要在model_data目录下创建一个类别文件voc_classes.txt
数据处理函数
通过上面代码会生成一个2007_train.txt和2007_test.txt文件,里面包含包含了图片的地址,和bbox框的坐标和类别。