记录mmdetection的安装过程

本文记录了使用Docker安装mmdetection的过程,包括如何修改代码、准备数据、训练模型等关键步骤。在Docker环境下,安装简便。安装后遇到的问题包括定位并修改库代码、将labelme标注转换为VOC format、调整配置文件以适应训练需求。模型训练前需修改VOC.py的Classes、config files的相关参数,并更新class_name.py。
摘要由CSDN通过智能技术生成

mmdetection 是香港中文大学Multimedia Laboratory基于Pytorch开发的物体识别的工具箱。然后讲述一下具体的安装和使用过程防止遗忘。
我是使用Docker安装的,十分的方便。

docker build -t mmdetection docker/

然后记录一下关键的几个坑。

修改代码

可能你在安装之前会clone下来mmdetection,但是实际上mmdetection作为一个库被安装在了某个地方,就像numpy这些库一样。所以需要修改代码请找到库的安装位置并修改。

数据准备

准备数据的时候我是使用的labelme,然后将标注文件转换成VOC format. 具体的format如下。

mmdetection
├── mmdet
├── tools
├── configs
├── data
│   ├── VOCdevkit
│   │   ├── VOC2007
│   │   │   ├── Annotations
│   │   │   ├── JPEGImages
│   │   │   ├── ImageSets
│   │   │   │   ├── Main
│   │   │   │   │   ├── test.txt
│   │   │   │   │   ├── trainval.txt

但是 labelme转换出来并不是直接就符合这样的格式,因此需要通过下面的代码去进行创建ImageSets目录以及相关文件。

import os  
import random  

if __name__=="__main__":
  
    trainval_percent = 0.80  
    train_percent &
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值