mmdetection安装

文章目录

  • 前言
  • 一、mmdetection是什么?
  • 二、mmdetection的安装
  • 总结


前言

这里为大家整理下自己使用mmdetection的一些经验

一、mmdetection是什么?

mmdetection是商汤科技(2018 COCO 目标检测挑战赛冠军)和香港中文大学最近开源了一个基于Pytorch实现的深度学习目标检测工具箱。

二、mmdetection的安装

安装依赖的过程中一定要考虑MMDetection 和 MMCV 版本兼容性
下面给的链接里有一个表格,里面有所有mmdet对应的mmcv版本。简单的按照链接里的教程走,一般都不会有问题。里面还有更加详细的教程

# 如果官网下载比较慢的话,可以自己先在自己创建的虚拟环境下安装好GPU版本的pytorch,然后在自己的虚拟环境下执行pip install openmim后的内容
conda create -n openmmlab python=3.7 pytorch==1.6.0 cudatoolkit=10.1 torchvision -c pytorch -y 
conda activate openmmlab
pip install openmim
mim install mmcv-full
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .   #注意这里结尾有个点

转中文教程链接:里面有多种方法https://mmdetection.readthedocs.io/zh_CN/latest/get_started.html#id2

总结

安装有什么问题,可以在评论区留言。

你可能感兴趣的:(pytorch,深度学习,人工智能,目标检测)