欢迎光临散文网 会员登陆 & 注册

目标检测算法丨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框的坐标和类别。

目标检测算法丨YOLOv3数据处理的评论 (共 条)

分享到微博请遵守国家法律