文章目录

  • 1. 总体环境要求
  • 2. 安装步骤
    • 2.1 基础软件安装
      • 2.1.1 安装vs2017
      • 2.1.2 安装显卡驱动(windows系统如自带,则不必安装)
      • 2.1.3 安装cuda10
      • 2.1.4 安装cudnn7.6.4(以下以cudnn7.5为例)
      • 2.1.5 安装anaconda3
      • 2.1.6 安装gitbash
    • 2.2 基础代码库下载与编译
      • 2.2.1 anaconda虚拟环境建立并安装pytorch
      • 2.2.2 mmcv下载与调整
      • 2.2.3 mmdetection下载与调整
      • 2.2.4 SOLO下载与调整
  • 3 测试

1. 总体环境要求

  • cuda 9.2 硬性要求,否则失败,torch1.2.0推荐cuda-10.0/10.1,cudnn-7.6.4
  • torch 1.2.0, torchvision 0.4.0
  • C ++ 2015及以上
  • mmdetection 1.1.0
  • mmcv 新的可以用(依旧推荐mmcv==2.0.16)
  • SOLOV2 新的可以用
  • opencv-python 新的可以用

2. 安装步骤

2.1 基础软件安装

2.1.1 安装vs2017

下载安装选择c++即可:

2.1.2 安装显卡驱动(windows系统如自带,则不必安装)

显卡驱动:nvidia显卡驱动
选择对应的版本和显卡型号:
下载驱动后,默认一直下一步即可:



2.1.3 安装cuda10

下载地址:cuda10
安装,选择自定义后,一直下一步。



安装完成后,检查环境变量中是否添加成功:

测试cuda是否能运行:
cmd中运行nvcc -V
能打印出信息则安装成功

2.1.4 安装cudnn7.6.4(以下以cudnn7.5为例)

下载cudnn需要登陆nvidia账号:
下载cudnn7.6.4:cudnn7.6.4


下载解压后,按照下面的表,把对应文件复制过去:

添加环境变量:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64

2.1.5 安装anaconda3

下载地址:anaconda

下载后,双击安装即可。

anaconda3安装注意事项参见:Mask-RCNN应用 - Win10上安装Mask-RCNN运行环境- 第二步:安装Anaconda

2.1.6 安装gitbash

用于下载github上的项目和安装pycocotools
下载地址:gitbash

2.2 基础代码库下载与编译

conda国内使用清华源加速:conda清华源
pip使用清华源:pypi清华源
一定要管理员权限运行:

2.2.1 anaconda虚拟环境建立并安装pytorch

conda create -n solo python=3.7
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

如果由于网络原因pytorch按照上述方法下载较慢,可通过离线安装包安装,安装包下载链接为:
链接:https://pan.baidu.com/s/19RXQdrQ9kgyV-P3mKBNAog
提取码:rb97

2.2.2 mmcv下载与调整

mmcv 新的可以用(依旧推荐mmcv==2.0.16)

git clone https://github.com/open-mmlab/mmcv
cd mmcv
pip install .
pip install Cython==0.29.14

2.2.3 mmdetection下载与调整

mmdetection 1.1.0

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements.txt
# 进行编译
python setup.py build_ext --inplace
# 完成安装
python setup.py install develop

安装成功后查看:pip list

2.2.4 SOLO下载与调整

SOLOV2 新的可以用
注意以下第二步: pycocotool/cocoapi的安装需要注意,参考Mask-RCNN应用 - Win10上安装Mask-RCNN运行环境-第六步:基于mask rcnn的其他需要安装部分-3.pycocotools 安装

git clone https://github.com/WXinlong/SOLO.git
cd SOLO
pip install -r requirements/build.txt
pip install "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI"
pip install -v -e .  # or "python setup.py develop"

3 测试

该代码在SOLO/demo文件夹中可以找到
代码中需要的checkpoints/DECOUPLED_SOLO_R50_3x.pth与config文件中的py文件相对应
checkpoints文件夹需要自己建立并下载对应模型,模型下载链接如文末

笔者在测试时发现当前代码运行有bug,需要修改源码,如需要该源码或学习指导,请用手机打开以下链接:https://m.tb.cn/h.fINaraE?tk=PCzA2jPp4V0,进行咨询

from mmdet.apis import init_detector, inference_detector, show_result_pyplot, show_result_ins
import mmcvconfig_file = '../configs/solo/decoupled_solo_r50_fpn_8gpu_3x.py'
# download the checkpoint from model zoo and put it in `checkpoints/`
checkpoint_file = '../checkpoints/DECOUPLED_SOLO_R50_3x.pth'# build the model from a config file and a checkpoint file
model = init_detector(config_file, checkpoint_file, device='cuda:0')# test a single image
img = 'demo.jpg'
result = inference_detector(model, img)show_result_ins(img, result, model.CLASSES, score_thr=0.25, out_file="demo_out.jpg")

如测试没有问题,则在demo文件夹下会生成如下图所示的“demo_out.jpg”

models:

Model Multi-scale training Testing time / im AP (minival) Link
SOLO_R50_1x No 77ms 32.9 download
SOLO_R50_3x Yes 77ms 35.8 download
SOLO_R101_3x Yes 86ms 37.1 download
Decoupled_SOLO_R50_1x No 85ms 33.9 download
Decoupled_SOLO_R50_3x Yes 85ms 36.4 download
Decoupled_SOLO_R101_3x Yes 92ms 37.9 download
SOLOv2_R50_1x No 54ms 34.8 download
SOLOv2_R50_3x Yes 54ms 37.5 download
SOLOv2_R101_3x Yes 66ms 39.1 download
SOLOv2_R101_DCN_3x Yes 97ms 41.4 download
SOLOv2_X101_DCN_3x Yes 169ms 42.4 download

Light-weight models:

Model Multi-scale training Testing time / im AP (minival) Link
Decoupled_SOLO_Light_R50_3x Yes 29ms 33.0 download
Decoupled_SOLO_Light_DCN_R50_3x Yes 36ms 35.0 download
SOLOv2_Light_448_R18_3x Yes 19ms 29.6 download
SOLOv2_Light_448_R34_3x Yes 20ms 32.0 download
SOLOv2_Light_448_R50_3x Yes 24ms 33.7 download
SOLOv2_Light_512_DCN_R50_3x Yes 34ms 36.4 download

SOLO在windows 10环境下安装相关推荐

  1. 在windows 10环境下安装 mujoco

    陆陆续续,花了一个星期,把各个模块翻了一遍,踩了无数坑.为了弄清楚中间的调用关系,连_msvccompiler.py,core.py这些都被我改过.最后,回归简单,按照正确的思路和方案安装,就没问题. ...

  2. windows 10环境下安装Tesseract-OCR与python集成

    前言 Tesseract是一个开源的ocr引擎,可以开箱即用,项目最初由惠普实验室支持,1996年被移植到Windows上,1998年进行了C++化.在2005年Tesseract由惠普公司宣布开源. ...

  3. 【Linux】在Windows 10环境下安装适用于 Linux 的子系统 (WSL安装指南)

    更多教程:https://docs.microsoft.com/zh-cn/windows/wsl/install-win10 安装适用于 Linux 的 Windows 子系统 (WSL) 时有两个 ...

  4. tensor搭建--windows 10 64bit下安装Tensorflow+Keras+VS2015+CUDA8.0 GPU加速

    windows 10 64bit下安装Tensorflow+Keras+VS2015+CUDA8.0 GPU加速 原文见于:http://www.jianshu.com/p/c245d46d43f0 ...

  5. Windows 10系统下安装MATLAB出现Error 1935的应对措施

    Windows 10系统下安装MATLAB出现Error 1935的应对措施 出现该错误的原因是安装过程默认安装的时C++ 2005而不是2008,因此绕开C++ 2005来安装.具体步骤如下: 1. ...

  6. Windows 10环境中安装Snort+Barnyard2+MySQL

    一.背景 Snort是一款著名的开源入侵检测系统,有着悠久的历史和众多用户.我们通过部署Snort系统来监测异常网络活动并发出报警.Snort的跨平台性能很好,它支持的各种Linux/Unix和Win ...

  7. windows 10 64bit下安装Tensorflow+Keras+VS2015+CUDA8.0 GPU加速

    https://www.jianshu.com/p/c245d46d43f0 写在前面的话 2016年11月29日,Google Brain 工程师团队宣布在 TensorFlow 0.12 中加入初 ...

  8. 目标检测第3步:如何在Windows 10系统下安装CUDA(更新时间2022.03.22)

    (请先看置顶博文)本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客 目录 一.自检 1.查看PC是否有NVIDIA的独立显卡 2.查看自己NVIDIA独立显卡是否支持CUD ...

  9. Windows 10环境下TensorFlow(gpu版本)配置教程——[图解] [详细版][零基础]

    Tensorflow环境下的深度学习框架的配置主要包含以下几步: 0.前言 1.PyCharm的安装步骤: 2.Python的安装步骤: 3.AnaConda的安装步骤: 4.CUDA的安装步骤: 5 ...

最新文章

  1. 会数据分析的人别再低调了,我怕你会因此错失100万奖金
  2. python实现人脸识别抓取人脸并做成熊猫头表情包(2)之优化
  3. hdu1358 最小循环节,最大循环次数 KMP
  4. CF1101D GCD Counting
  5. python3爬虫(5)百度云盘暴力破解尝试
  6. C学习杂记(二)笔试题:不使用任何中间变量如何将a、b的值进行交换
  7. 软件项目可行性分析定义_如何定义最低可行产品
  8. Storm精华问答 | storm与Hadoop区别?
  9. Mysql windows安装说明
  10. [cb]SceneView 获取鼠标位置
  11. 4、IO--字节转换流
  12. warning: mysql-community-libs-5.7.11-1.el7.x86_64.rpm: Header V3 DSA/SHA1 Signature, key ID 5072e1f5
  13. PHP两个二维数组数据处理:字段值相等,则赋值给另一个数组
  14. 计算机基础制作幻灯片讲解,计算机应用基础_幻灯片制作.ppt
  15. 第三阶段应用层——1.3 数码相册—英文和汉字的点阵显示
  16. math.sqrt java_Java Math sqrt() 使用方法及示例
  17. 安装CDH6.3版本的时候遇到 ‘ERROR main:com.cloudera.server.cmf.Main: Server failed’的解决方案
  18. 实用Windows网速监控软件TrafficMonitor
  19. 360 支持linux版本下载地址,360安全卫士linux版下载
  20. 全国省份、城市关联表 mysql(含城市名拼音)

热门文章

  1. 雅克比矩阵(上)-----雅克比推导
  2. 【面试必备】java开发转算法工程师
  3. golang通过递归遍历生成树状结构
  4. 如何把后端返还的blob格式的文件进行本地下载?
  5. 浅谈联邦学习Federated Learning
  6. 线性方程组的SOR迭代法
  7. Sql Server 数据类型
  8. 如何解决 conda install 库时报错:The environment is inconsistent, please check the package plan carefully
  9. 3.Matlab——数组大小
  10. 微软Surface系列固件命名规则