win11下Swin-Transformer-Object-Detection-master测试成功

已于 2022-05-06 02:10:29 修改
阅读量1.2k 收藏 8
点赞数 3
文章标签： transformer pytorch 深度学习
于 2022-05-05 12:53:41 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/impact_10/article/details/124586061
版权

仅作为个人学习记录使用，侵权直接删除！！！
转自这里（侵权删）
前期准备
PyCharm 2021.3（只是IDE，版本不同也没关系）
显卡 3060 12G
Cuda 11.1 （与显卡、Pytorch要求合适即可）
Python 3.8
Anaconda3 2021.05（版本无限制）
VS2019

环境安装
VS 2019 建议安装2019版本

  1. VS 2019 安装
    VS 2019下载地址

直接去VS官网下载，第一步下载的时候可能需要翻墙，后面下载的时候不用翻墙，安装时只勾选桌面端的C++部分就可以。

添加环境变量到系统环境变量->Path下，地址在安装目录下（根据自己的安装位置）：D:\install_in_D\VS2019\2019\Community\VC\Tools\MSVC\14.29.30133\bin\Hostx86\x64
验证是否可用：
在这里插入图片描述

  1. 创建conda 虚拟环境
    方案一（直接创建）
    直接创建：conda create -n swin python=3.8
    激活conda环境：activate swin

  2. pytorch 安装（版本1.8.2）
    官网地址：https://pytorch.org/

安装：pip3 install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

  1. 安装其他必要的Python包
    在虚拟环境下执行：
pip install cython matplotlib opencv-python timm -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
  1. 安装mmcv
    项目地址：mmcv
    下载好mmcv之后，进行接下来的安装操作
    打开 conda的power shell（一定要用这个）
    在这里插入图片描述
    进入到 conda环境目录，激活conda 环境
    直接创建的conda环境的使用：activate swin
    进入到 mmcv 目录下，执行
pip install -r requirements.txt

来安装mmcv所需的python包

配置环境变量：因为在安装mmcv时用到。
在这里插入图片描述
在这个位置进行新建环境变量
第一步：
$env:TORCH_CUDA_ARCH_LIST="8.0" 8.0为显卡算力
在这里插入图片描述
第二步：

$env:MMCV_WITH_OPS = 1 #

在这里插入图片描述
第三步：
$env:MAX_JOBS = 8 根据CPU核心数设置（其实我是10核心，不大于你CPU核心应该就没事）
在这里插入图片描述

确保vs安装成功并且有C++模块
运行：python setup.py build_ext（这步包括下面的一步就会用到之前VS中的cl工具）
运行：python setup.py develop
验证：pip list 会看到 mmcv-full（记得确定一下版本对不对 1.3.17）
在这里插入图片描述
安装mmcv-full通常会出现问题，一定要注意vs安装成功后是否重启电脑，不重启电脑会导致vs无法正常使用，代码运行自然也是错误的！（亲身经历）
通常情况下mmcv-full默认的安装版本为1.5.0但是在本次代码中使用的是1.3.17，版本太高是不能正常运行代码的。切记！切记！切记！
可以在配置好前面的操作后，使用下面的指令进行安装：

pip install mmcv-full==1.3.17 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.2/index.html

这是本文中使用到的指令！

  1. 安装mmdetection

激活虚拟环境activate swin
克隆mmdetection 的代码，找到要拉取的版本后直接下载就行，地址：https://github.com/open-mmlab/mmdetection/tree/v2.20.0
进入 mmdetection的代码目录，安装mmdetection所需的python包，执行：pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com (会在下载mmtrack时候出错，这是某个测试环境需要的包，暂时不管它，把它屏蔽掉，再运行上面命令)。
在这里插入图片描述
安装 mmdetection，执行 ：python setup.py develop
验证 pip list 可以看到 mmdet（到这一步的时候版本为 2.20.0，但是后面配置Transformer时候会自动变成2.11.0，这个无伤大雅）
在这里插入图片描述

  1. 安装 apex
    注：apex可以尝试直接用pip装，但是不是确定是否可用，可以尝试。
    apex 项目：https://github.com/NVIDIA/apex
    进入 apex 文件夹
    执行：python setup.py install
    pip list 能看见 apex （0.1版本，只有这一个版本）
    注：安装的apex会在训练模型时候有一个警告内容如下：（但实际没啥影响）–》个人没有遇到这个警告
fused_weight_gradient_mlp_cuda module not found. gradient
 accumulation fusion with weight gradient computation disabled.
  1. 拉取Swin-Transformer-Object-Detection-master代码
    地址：Swin-Transformer-Object-Detection-master
    所用环境为前面准备好的虚拟环境
    进入虚拟环境后配置Swin-Transformer-Object-Detection-master所需的一些补充环境：
    运行pip install -r requirements.txt
    在虚拟环境下，进入Swin-Transformer-Object-Detection 项目目录，执行：python setup.py develop
    到这里基本没什么问题了
  2. 测试工程

下载权值文件：(直接丢在项目根目录下)
在这里插入图片描述
验证执行如下命令：

python demo/image_demo.py demo/demo.jpg configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py mask_rcnn_swin_tiny_patch4_window7.pth

成功的话 能看到测试成功的实例图片
在这里插入图片描述
温馨提示：一定要一个步骤一个步骤的走，能避免绝大部分的问题，显卡算力可以适当的降低为自己显卡低一级别的算力

impact_10
关注
  • 3
    点赞
  • 踩
  • 8
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
Swin-Transformer.zip
12-06
Swin-Transformer图像分类实战，详见文章：https://blog.csdn.net/hhhhhhhhhhwwwwwwwwww/article/details/121744503?spm=1001.2014.3001.5501
win10配置mmcv
weixin_46463800的博客
10-26 393
显卡：NVIDIA GeForce RTX 3070 Laptop GPU conda create -n mmseg python=3.7 conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch pip install torch1.9.0 pip install torchaudio0.9.0 pip install torchvision==0.10.0 在环境变量配置好的情况下 $env:TORCH_CUDA_
Swin-Transformer-Object-Detection环境搭建(Win10+Anaconda3+mmdetection)(最详细)
热门推荐
畅游计算机视觉的海洋
04-30 20万+
原文地址：https://arxiv.org/abs/2103.14030 代码地址：https://github.com/microsoft/Swin-Transformer 目标检测：https://github.com/SwinTransformer/Swin-Transformer-Object-Detection mmdetection项目地址：mmdetection.getstarted 文章目录前言1.mmdetection版本要求2.Swin-Transformer-Object-Detec
swin transformer环境配置（矩池云）
m0_51709155的博客
03-06 1363
该篇文章主要帮助小白快速完成swin transformer环境配置
字符识别-基于Swin-Transformer实现的OCR字符识别算法-附项目源码-优质项目实战.zip
最新发布
04-29
在这个名为“字符识别-基于Swin-Transformer实现的OCR字符识别算法-附项目源码-优质项目实战.zip”的压缩包中，包含了一个高质量的实践项目，该项目专注于使用Swin-Transformer这一先进的深度学习模型来实现OCR...
TensorRT-使用tensorrt部署Swin-Transformer-支持fp16+int8精度推理-优质算法部署项目实战
02-28
在本项目实战中，我们将深入探讨如何利用NVIDIA的TensorRT框架来部署Swin-Transformer模型，这是一种先进的计算机视觉算法，特别适用于图像分类、目标检测等任务。TensorRT是一个高性能的深度学习推理（Inference）...
Swin-Transformer+gradcam可视化代码+Windows环境.zip
01-18
Swin-Transformer+gradcam可视化代码+Windows环境
Swin-T-使用Pytorch实现Swin-Transformer目标检测算法-优质项目实战.zip
04-29
3. **目标检测框架**：Swin-Transformer可以与多种目标检测框架集成，如MMDetection或Detectron2。项目中会介绍如何将Swin-Transformer整合到这些框架中，替换原有的backbone网络，如ResNet或EfficientNet。 4. **...
Swin-Transformer（code and 论文）.zip
12-29
Transformer以其自注意力机制在自然语言处理(NLP)中取得了巨大成功，Swin-Transformer则是将这种机制应用于图像处理，创建了一个层次化的、窗口化的Transformer模型。 论文“Swin Transformer: Hierarchical Vision...
【SWIN-Transformer环境配置及个人VOC数据集的训练】
weixin_42686221的博客
06-26 1613
  本文对在Windows系统下配置mmdetection环境并通过SWIN+MASK R-CNN网络对个人的VOC格式的数据集过程中遇到的问题进行总结，偏向于是对一些问题解答博客的汇总。  环境配置我参考了下文博客: https://blog.csdn.net/CSDN_X_W/article/details  使用annaconda安装，pytorch安装的版本是1.8.0，cuda安装的是11.1...
Swin Transformer实战目标检测：训练自己的数据集
01-04
Transformer发轫于NLP（自然语言处理），并跨界应用到CV（计算机视觉）领域。 Swin Transformer是基于Transformer的计算机视觉骨干网，在图像分类、目标检测、实例分割、语义分割等多项下游CV应用中取得了SOTA的性能。该项工作也获得了ICCV 2021顶会最佳论文奖。 本课程将手把手地教大家使用labelImg标注和使用Swin Transformer训练自己的数据集。  本课程将介绍Transformer及在CV领域的应用、Swin Transformer的原理。 课程以多目标检测（足球和梅西同时检测）为例进行Swin Transformer实战演示。 课程在Windows和Ubuntu系统上分别做项目演示。包括：安装软件环境、安装Pytorch、安装Swin-Transformer-Object-Detection、标注自己的数据集、准备自己的数据集（自动划分训练集和验证集）、数据集格式转换（Python脚本完成）、修改配置文件、训练自己的数据集、测试训练出的网络模型、性能统计、日志分析。  相关课程： 《Transformer原理与代码精讲（PyTorch）》https://edu.csdn.net/course/detail/36697《Transformer原理与代码精讲（TensorFlow）》https://edu.csdn.net/course/detail/36699《ViT（Vision Transformer）原理与代码精讲》https://edu.csdn.net/course/detail/36719《DETR原理与代码精讲》https://edu.csdn.net/course/detail/36768《Swin Transformer实战目标检测：训练自己的数据集》https://edu.csdn.net/course/detail/36585《Swin Transformer实战实例分割：训练自己的数据集》https://edu.csdn.net/course/detail/36586《Swin Transformer原理与代码精讲》 https://download.csdn.net/course/detail/37045
yolov5-7.0简单训练教程
weixin_45185577的博客
02-20 2159
yolov5-7.0教程。以上就是yolov5-7.0的简单训练教程啦，希望大家都可以成功。
Swin Transformer实战实例分割：训练自己的数据集
01-04
Transformer发轫于NLP（自然语言处理），并跨界应用到CV（计算机视觉）领域。 Swin Transformer是基于Transformer的计算机视觉骨干网，在图像分类、目标检测、实例分割、语义分割等多项下游CV应用中取得了SOTA的性能。该项工作也获得了ICCV 2021顶会最佳论文奖。本课程将手把手地教大家使用labelme标注和使用Swin Transformer训练自己的数据集进行图片和视频的实例分割。  本课程将介绍Transformer及在CV领域的应用、Swin Transformer的原理。 本课程以汽车驾驶场景图片和视频开展项目实践：对汽车行驶场景中的路坑、车、车道线进行物体标注和实例分割。  课程在Windows和Ubuntu系统上分别做项目演示。包括：安装软件环境、安装Pytorch、安装Swin-Transformer-Object-Detection、标注自己的数据集、准备自己的数据集、数据集格式转换（Python脚本完成）、修改配置文件、训练自己的数据集、测试训练出的网络模型、性能统计、日志分析。  本课程提供项目的数据集和相关Python程序文件。相关课程： 《Transformer原理与代码精讲（PyTorch）》https://edu.csdn.net/course/detail/36697《Transformer原理与代码精讲（TensorFlow）》https://edu.csdn.net/course/detail/36699《ViT（Vision Transformer）原理与代码精讲》https://edu.csdn.net/course/detail/36719《DETR原理与代码精讲》https://edu.csdn.net/course/detail/36768《Swin Transformer实战目标检测：训练自己的数据集》https://edu.csdn.net/course/detail/36585《Swin Transformer实战实例分割：训练自己的数据集》https://edu.csdn.net/course/detail/36586《Swin Transformer原理与代码精讲》 https://download.csdn.net/course/detail/37045
安装mmcv，测试文档时，出现报错
qq_43199049的博客
05-11 445
E:\mini\envs\lane_seg\lib\site-packages\torch\utils\cpp_extension.py:287: UserWarning: Error checking compiler version for cl: [WinError 2] 系统找不到指定的文件。解决方法一、： 安装cmake工具，解压放在虚拟环境，包的文件夹中，解决方法二、： 安装cmake工具，解压放在虚拟环境，包的文件夹中，
Windows10下面安装MMCV全过程图文详解
u014723479的博客
10-30 8465
https://github.com/open-mmlab/mmcv 简介 是OpenMMLab家自己开发的模块、包，其他框架的运行，例如MMDetection 的运行是依赖于这个库的 MMCV 是一个面向计算机视觉的基础库，它支持了很多开源项目，例如： MMClassification: OpenMMLab 图像分类工具箱 MMDetection: OpenMMLab 目标检测工具箱 MMDetection3..
Swin-Transformer-Object-Detection项目安装教程
m0_56192771的博客
07-21 2119
Swin-Transformer-Object-Detection项目安装教程
Windown10 配置Swin-Transformer-Object-Detectio目标检测环境
weixin_62269220的博客
10-13 446
将文件放入C:\Users\nd\AppData\Local\miniconda3\envs\swindet。遇到的最大问题是各个包之间的版本不兼容的问题，耗时2天完成环境的配置。下载，解压同样放到你创建的环境中的文件夹。将权值文件放在你的swin解压的文件夹下。这时文件中就有mmcv的文件夹。查看conda环境位置 命令。CUDA版本11.8。
笔记--Win11安装mmcv-full
牵一只蜗牛去散步
03-11 2717
1--前言 电脑原配置： ①系统：Win11家庭中文版 ②Cuda版本：11.4.158（安装将改用10.2的版本） ③显卡：RTX3060 2--安装前的工作 2-1--安装10.2的Cuda版本 NVIDIA CUDA驱动版本下载地址：​​​​​​https://developer.nvidia.com/cuda-toolkit-archive cudnn下载地址 ：https://developer.nvidia.com/rdp/cudnn-archive 注1：安装Cuda10
swin-transformer概念
09-16
swin-transformer是一种用于计算机视觉任务的通用主干网络。它可以被应用于图像分类、图像分割、目标检测等多个视觉下游任务。swin-transformer通过引入窗口位移机制，将输入图像分割成多个局部区域，并在每个局部...
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值