SOLO在windows 10环境下安装
文章目录
- 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环境下安装相关推荐
- 在windows 10环境下安装 mujoco
陆陆续续,花了一个星期,把各个模块翻了一遍,踩了无数坑.为了弄清楚中间的调用关系,连_msvccompiler.py,core.py这些都被我改过.最后,回归简单,按照正确的思路和方案安装,就没问题. ...
- windows 10环境下安装Tesseract-OCR与python集成
前言 Tesseract是一个开源的ocr引擎,可以开箱即用,项目最初由惠普实验室支持,1996年被移植到Windows上,1998年进行了C++化.在2005年Tesseract由惠普公司宣布开源. ...
- 【Linux】在Windows 10环境下安装适用于 Linux 的子系统 (WSL安装指南)
更多教程:https://docs.microsoft.com/zh-cn/windows/wsl/install-win10 安装适用于 Linux 的 Windows 子系统 (WSL) 时有两个 ...
- 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 ...
- Windows 10系统下安装MATLAB出现Error 1935的应对措施
Windows 10系统下安装MATLAB出现Error 1935的应对措施 出现该错误的原因是安装过程默认安装的时C++ 2005而不是2008,因此绕开C++ 2005来安装.具体步骤如下: 1. ...
- Windows 10环境中安装Snort+Barnyard2+MySQL
一.背景 Snort是一款著名的开源入侵检测系统,有着悠久的历史和众多用户.我们通过部署Snort系统来监测异常网络活动并发出报警.Snort的跨平台性能很好,它支持的各种Linux/Unix和Win ...
- windows 10 64bit下安装Tensorflow+Keras+VS2015+CUDA8.0 GPU加速
https://www.jianshu.com/p/c245d46d43f0 写在前面的话 2016年11月29日,Google Brain 工程师团队宣布在 TensorFlow 0.12 中加入初 ...
- 目标检测第3步:如何在Windows 10系统下安装CUDA(更新时间2022.03.22)
(请先看置顶博文)本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客 目录 一.自检 1.查看PC是否有NVIDIA的独立显卡 2.查看自己NVIDIA独立显卡是否支持CUD ...
- Windows 10环境下TensorFlow(gpu版本)配置教程——[图解] [详细版][零基础]
Tensorflow环境下的深度学习框架的配置主要包含以下几步: 0.前言 1.PyCharm的安装步骤: 2.Python的安装步骤: 3.AnaConda的安装步骤: 4.CUDA的安装步骤: 5 ...
最新文章
- 会数据分析的人别再低调了,我怕你会因此错失100万奖金
- python实现人脸识别抓取人脸并做成熊猫头表情包(2)之优化
- hdu1358 最小循环节,最大循环次数 KMP
- CF1101D GCD Counting
- python3爬虫(5)百度云盘暴力破解尝试
- C学习杂记(二)笔试题:不使用任何中间变量如何将a、b的值进行交换
- 软件项目可行性分析定义_如何定义最低可行产品
- Storm精华问答 | storm与Hadoop区别?
- Mysql windows安装说明
- [cb]SceneView 获取鼠标位置
- 4、IO--字节转换流
- warning: mysql-community-libs-5.7.11-1.el7.x86_64.rpm: Header V3 DSA/SHA1 Signature, key ID 5072e1f5
- PHP两个二维数组数据处理:字段值相等,则赋值给另一个数组
- 计算机基础制作幻灯片讲解,计算机应用基础_幻灯片制作.ppt
- 第三阶段应用层——1.3 数码相册—英文和汉字的点阵显示
- math.sqrt java_Java Math sqrt() 使用方法及示例
- 安装CDH6.3版本的时候遇到 ‘ERROR main:com.cloudera.server.cmf.Main: Server failed’的解决方案
- 实用Windows网速监控软件TrafficMonitor
- 360 支持linux版本下载地址,360安全卫士linux版下载
- 全国省份、城市关联表 mysql(含城市名拼音)