MMDetection训练自己的数据集_在mmdetection中使用retinanet训练自己的数据集

MMDetection

1. 安装

1.1 windows

• 1.1.1 克隆git仓库

  git clone https://github.com/open-mmlab/mmdetection.git
  1

• 1.1.2 创建环境

  conda create --name openmmlab python=3.8 -y
  conda activate openmmlab
  conda install pytorch torchvision -c pytorch
  pip install -U openmim
  mim install mmcv-full
  git clone https://github.com/open-mmlab/mmdetection.git
  cd mmdetection
  pip install -v -e .
  # "-v" means verbose, or more output
  # "-e" means installing a project in editable mode,
  # thus any local modifications made to the code will take effect without reinstallation.
  1
  2
  3
  4
  5
  6
  7
  8
  9
  10
  11

• 1.1.3 验证

  mim download mmdet --config yolov3_mobilenetv2_320_300e_coco --dest .
  python demo/image_demo.py demo/demo.jpg yolov3_mobilenetv2_320_300e_coco.py yolov3_mobilenetv2_320_300e_coco_20210719_215349-d18dff72.pth --device cpu --out-file result.jpg
  1
  2