Detectron训练自己的数据集(voc格式的数据集)笔记

本文详细介绍了如何使用Detectron训练自定义的VOC格式数据集,包括Detectron环境搭建、数据转换、目录结构设置、配置文件修改及模型训练的步骤。通过创建软链接、下载预训练模型并调整配置文件,最终成功启动训练过程。
摘要由CSDN通过智能技术生成

detectron是facebook的开源项目,其实现主要基于caffe2。里面有诸如FPN,Mask-RCNN等优秀目标检测算法的实现。
利用Detectron训练自己的数据集,首先我们要搭建好相应环境。
caffe2的安装请参考caffe2官网安装教程。
https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=prebuilt
detectron安装请参考detectron官方安装教程。
https://github.com/facebookresearch/Detectron/blob/master/INSTALL.md

数据准备
由于我们的数据集是voc格式的,即我们的annotation是xml格式的,而detectron要求的是json格式的annotation,因此首先进行数据转换。

首先将训练集的xml格式的annotation放置到新文件夹train_annotations, 将测试集的xml格式的annotations放置到新文件夹test_annotations。(可能我们的测试集是从训练集中分离出来的验证集,它们的xml格式的annotation放置在一起,手动分离是不太可能的,对此,我们可以用程序实现)

# if you want to use it to copy some fil
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值