mmdetection安装记录（使用google colab安装）

2020-03-03 11:42:57 阅读：1119 来源： 互联网

标签：gcc google colab ++ demo args && mmdetection 安装

介绍

Google Colaboratory是谷歌开放的一款研究工具，主要用于机器学习的开发和研究。它给广大的AI开发者提供了免费的GPU使用，可以在上面跑tensorflow等框架。

mmdetection集成了多种目标检测算法，它的git地址,环境要求在官方文档中。我在Google Colaboratory上搭建环境的时候遇到了问题，记录如下。

安装过程

安装成功的环境

OS: Ubuntu18.04
CUDA:10.0
NCCL: 1.3.5
GCC(G++): 5.5

具体安装步骤

1.安装Anaconda3

!wget -q https://repo.anaconda.com/archive/Anaconda3-2019.10-Linux-x86_64.sh
!chmod +x Anaconda3-2019.10-Linux-x86_64.sh
!bash ./Anaconda3-2019.10-Linux-x86_64.sh -b -f -p /content/anaconda3
# set PATH environment variable
import os
os.environ['PATH'] = "/content/anaconda3/bin:" + os.environ['PATH']

2.安装gcc(g++5)

由于系统自带的gcc是为7.4.0，需要替换成5.5版本

!apt-get install gcc-5 g++-5
!cd /usr/bin && sudo rm gcc-7 g++-7 gcc g++ && sudo ln gcc-5 gcc && sudo ln g++-5 g++

3.安装pytorch和cudatoolkit

!conda install cython
!conda install pytorch=1.3.1 torchvision cudatoolkit=10.0 nccl -c pytorch

4.验证gcc和cuda版本

!gcc --version
!g++ --version
!nvcc --version
!conda --version

5.安装mmdetect

!git clone https://github.com/open-mmlab/mmdetection.git
!cd mmdetection && pip install -r requirements/build.txt && pip install "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI" && python setup.py develop

6.测试环境

1.下载模型文件

!cd mmdetection && mkdir checkpoints && cd checkpoints && wget https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth

2.运行

代码中提供了webcam_demo,由于使用的是Colaboratory，没有摄像头，将result保存成图片，重写demo如下

import argparse
import cv2
import torch
from mmdet.apis import inference_detector, init_detector, show_result, show_result_pyplot

def parse_args():
    parser = argparse.ArgumentParser(description='MMDetection webcam demo')
    parser.add_argument('--config', default='../mmdetection/configs/faster_rcnn_r50_fpn_1x.py', help='test config file path')
    parser.add_argument('--checkpoint', default='../mmdetection/checkpoints/faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth',
                        help='checkpoint file')
    parser.add_argument('--device', type=int, default=0, help='CUDA device id')
    parser.add_argument(
        '--camera-id', type=int, default=0, help='camera device id')
    parser.add_argument(
        '--score-thr', type=float, default=0.5, help='bbox score threshold')
    args = parser.parse_args()
    return args

args = parse_args()
model = init_detector(
        args.config, args.checkpoint, device=torch.device('cuda', args.device))
img = '../mmdetection/demo/demo.jpg'
show_img = inference_detector(model, img)
show_result(img, show_img, model.CLASSES, score_thr=args.score_thr, show=False, out_file='test.jpg')

执行命令

!cd mmdetection && python demo/demo.py

报错如下：

ModuleNotFoundError: No module named 'cityscapesscripts'

安装cityscapesscripts

pip install cityscapesscripts

重新运行demo,成功结果如下

更改图片，再次运行

参考资料

用谷歌的GPU跑你的代码----Colaboratory使用记录

ubuntu18.04下安装mmdetection 及问题解决

关于ubuntu下mmdetection的安装步骤和心得

WhatPrime 发布了12 篇原创文章 · 获赞 7 · 访问量 9636 私信关注

标签：gcc,google,colab,++,demo,args,&&,mmdetection,安装
来源： https://blog.csdn.net/WhatPrime/article/details/104627098

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9