PaddleDetection支持43种数据格式,文中主要说明如何用自定义COCO和VOC数据集进行训练。新版本数据配置抽离,用户关注configs/datasets下的配置文件即可。使用自定义VOC数据集改voc.yml,改dataset_dir等路径;COCO目标检测改cooc_detection.yml,实例分割改cooc_instance.yml,均需修改相关路径。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

PaddleDetection支持的数据格式
目前#PaddleDetection支持43种数据格式:coco voc widerface。在这里我们主要说明一下如何使用自定义COCO进行目标检测、实例分割;如何使用自定义VOC数据集进行目标检测。在PaddleDetection新的版本中,我们将数据配置抽离了出来,用户在自定义数据集进行训练的时候,只需要关注在configs/datasets文件中的数据配置文件即可。
1 使用自定义VOC数据集进行目标检测训练
当用户使用VOC数据集进行的,只需要对voc.yml进行修改即可实现正常训练。我们提供了一个自定义的VOC数据集,并整理成如图如下图所示的形式。用户在voc.yml文件中修改路径即可。dataset_dir: 表示数据集的路径(该路径下保存数据集、标注文件、标签以及训练验证数据集说明txt文件)anno_path: 表示训练/验证数据集的说明txt文件路径 (该处填写相对于dataset_dir的相对路径)label_list: 表示label的txt文件路径(该处填写相对于dataset_dir的相对路径)在TestDataset的配置中,对于的主要是label_list的路径,确保在测试过程中,有正确的标签

2 使用自定义coco数据集进行目标检测训练
当用户使用COOC数据集进行目标检测的时候,只需要对cooc_detection.yml进行修改即可实现正常训练。
我们提供了一个自定义的COCO数据集,并整理成如图如下图所示的形式。用户在cooc_detection.yml文件中修改路径即可。
image_dir: 表示训练/验证图片的路径anno_path: 表示训练/验证图片的标注文件的路径dataset_dir: 表示数据集的路径(该路径下保存数据集、标签)在TestDataset的配置中,对于的主要是引入标签,这里会抽取val.json文件中标签,确保在测试过程中,有正确的标签

3 使用自定义COCO数据集进行实例分割训练
当用户使用COCO数据集进行实例分割的时候,只需要对cooc_instance.yml进行修改即可实现正常训练,基本何目标检测的coco数据集格式基本一致
我们提供了一个自定义的COCO数据集,并整理成如图如下图所示的形式。用户在cooc_instance.yml文件中修改路径即可。
image_dir: 表示训练/验证图片的路径anno_path: 表示训练/验证图片的标注文件的路径dataset_dir: 表示数据集的路径(该路径下保存数据集、标签)在TestDataset的配置中,对于的主要是引入标签,这里会抽取val.json文件中标签,确保在测试过程中,有正确的标签

以上就是【PaddleDetection2.0专项】如何自定义数据集的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/61791.html
微信扫一扫
支付宝扫一扫