Detectron安装与用GPU训练自己数据集(voc格式)

xiaoxiao2021-02-28  46

FAIR的Detectron平台是facebook基于coco数据集训练的,虽然官方建议使用coco格式数据集训练模型,但很多人习惯于用voc格式标记数据,也有现成的xml不知如何使用Detectron。本博文假设已经做好voc格式数据,稍做转换使其能够在Detectron上训练。本博文主要以resnet-50-fpn-faster-rcnn训练为例,只测试目标检测,说明如何使用自己的voc数据训练模型,关于FAIR训练好的模型demo部分不再叙述。

参考blog: https://blog.csdn.net/meccaendless/article/details/79457330

准备工作:

1、安装caffe2

参考blog: https://blog.csdn.net/yan_joy/article/details/70241319

(1) 安装caffe2依赖库

    sudo apt-get update     sudo apt-get install -y --no-install-recommends \       build-essential \       cmake \       git \       libgoogle-glog-dev \       libprotobuf-dev \  
转载请注明原文地址: https://www.6miu.com/read-2621838.html

最新回复(0)