【mmdetection】mmdetection安装详细步骤

         mmdetection是由商汤科技和香港中文大学开源了一个基于Pytorch实现的深度学习计算机视觉工具箱,涵盖了目标检测、实例分割、全景分割、模型蒸馏等计算机视觉任务,复现了最新的一些论文和成果,特别是包括了大量CVPR论文的复现。

1 项目地址

        项目地址:https://github.com/open-mmlab/mmdetection
        安装指南:https://github.com/open-mmlab/mmdetection/blob/master/docs/en/get_started.md

2 安装步骤

        mmdetection的安装步骤如下:

conda create -n openmmlab python=3.7 -y
conda activate openmmlab
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install openmim
mim install mmcv-full
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .

        其中,pytorch版本根据自身的cuda版本安装相应的版本即可。如果git无法使用,也可以用浏览器下载后解压。

3 效果测试

        (1)运行下面命令下载预训练模型

mim download mmdet --config yolov3_mobilenetv2_320_300e_coco --dest .

        运行完之后,在当前目录下会下载到两个文件,即yolov3_mobilenetv2_320_300e_coco.py 和yolov3_mobilenetv2_320_300e_coco_20210719_215349-d18dff72.pth。

        如果运行文件时,提示search.py的392行抛出如下异常:

        KeyError: 'Cascade Mask R-CNN'

        那么,修改python环境中/lib/python3.7/site-packages/mim/commands/search.py文件,这个文件在抱错时也会提示392行抛出异常,修改390行为:

if collection_name and collection_name in name2collection.keys():

        (2)运行测试脚本

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

        运行完成之后,文件夹下会出现目标检测结果图片result.jpg,如下图所示。

【mmdetection】mmdetection安装详细步骤_第1张图片

4 【python三维深度学习】python三维点云从基础到深度学习_Coding的叶子的博客-CSDN博客_python三维点云重建

更多三维、二维感知算法和金融量化分析算法请关注“乐乐感知学堂”微信公众号,并将持续进行更新。

你可能感兴趣的:(Pytorch,深度学习环境,python,mmdetection,pytorch,python,人工智能,深度学习)