Windows10系统下swin-transformer目标检测环境搭建_第1页
Windows10系统下swin-transformer目标检测环境搭建_第2页
Windows10系统下swin-transformer目标检测环境搭建_第3页
全文预览已结束

下载本文档

版权说明:本文档由用户提供并上传,收益归属内容提供方,若内容存在侵权,请进行举报或认领

文档简介

1、Windows10系统下swin-transformer标检测环境搭建Swin-Transformer-Object-Detection 环境搭建前环境总览:Windows10/11+ Microsoft Visual Studio2019 环境总览:Windows10/11+ Microsoft Visual Studio2019 社区版python3.7+pytorch1.8.1+torchvision0.9.1+cuda10.1mmcv1.3.8+mmdet2.11.0环境准备(1)安装 Visual Studio需要到 C+ 安装编译库,所以需要安装 Microsoft Visual

2、Studio,版本的话较常见的是2019,最新的2022版本应该也可以,安装很简单。下载地址:安装时记住安装位置,然后在 系统变量path 中添加VS2019的cl.exe件路径:*:Program Files (x86)Microsoft Visual Studio2019CommunityVCToolsMSVC14.*binHostx86x64(星号的位置根据的安装情况进替换 ),然后上移到最顶端,路确定之后退出,关机重启。(2)程项件准备checkpoints的件夹,将官公开的那些预训练模型放进去,整体就是这个样的。虚拟环境创建conda create -n swinmm python

3、conda create -n swinmm python =3.7 -yconda activate swinmmconda activate swinmmPS: swinmm为环境名称,可更改pytorch参考上篇的安装和运代码经验,接上步键:conda install pytorch conda install pytorch =1.8.1 torchvision =0.9.1 cudatoolkit=10.1 -c pytorchmmcv三种式可以选择,尝试哪种可以安装成功,这是个坑!(1)动安装(不稳定不推荐)pip install openmim mim install mmdet

4、pip install openmim mim install mmdet(2)动安装(强烈推荐)pip install mmcv pip install mmcv -full=1.3.8 -f https:/download.openmmlab .com/mmcv/dist/cu101/torch1.8.0/index.htmlPS:one:命令中的 cu101 和 torch1.8.0 为需要注意的地,不是按照上述步骤的话更改,这是1.8.0不是1.8.1的原因官有说明;three:这指定1.3.8版本是因为开始我第种法没有成功,然后的第三种法从源码安装,结果装的 mmcv 是最新版本1.

5、4.6,然后代码运是会报错,提:AssertionError: MMCV=1 AssertionError: MMCV=1 .4.6 is used but incompatible . Please install mmcv=1 .2.4, =1.4.0.所以mmcv的版本应该在1.2.4到1.4.0之间选择!git clone https :/git clone https :/open-mmlab/mmcv.git或者在该 github 仓库将 code 打包下载然后解压,然后在 Anaconda Powershell Prompt 中使 cd 命令切换到该根录下,然后:#build#b

6、uildpython setup .py build_ext# installpython setup .py developmmdet继续使 cd 命令将路径切换到第步准备的 Swin-Transformer-Object-Detection 程件路径下,如我的就是:cd Ecd E:/cd Swin-Transformer -Object-Detection继续键:pip install -r requirements pip install -r requirements .txtpip install -r requirements /docs.txtpython setup .py

7、developapex如果训练 swin 的话还需要到个叫 apex 的东西,安装步骤同样参考上篇中的第5步安装。字步骤如下,详细见原博客:测试PycharmSwin-Transformer-Object-Detection5步应该装的差不多了),然后新建demo.py 件,将以下代码复制进去:fromfrom mmdet.apis import init_detector,inference_detector,show_result_pyplotimport cv2# configs的路径可以按照这个寻找config_file = E:/Swin-Transformer-Object-Det

8、ection/configs/swin/cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py# 这个就是权重件,也就是第步环境准备中新建heckpoints件夹checkpoint_file = E:/Swin-Transformer-Object-Detection/checkpoints/cascade_mask_rcnn_swin_tiny_patch4_window7.pth device = cuda:0# init a detectormodel = init_detector (config_file, checkpoint_file, device=device)# inference the demo image# 测试图在官程件demo件夹下image = E:/Swin-Transformer-Object-Detection/demo/demo.jpgresult = inference

温馨提示

  • 1. 本站所有资源如无特殊说明,都需要本地电脑安装OFFICE2007和PDF阅读器。图纸软件为CAD,CAXA,PROE,UG,SolidWorks等.压缩文件请下载最新的WinRAR软件解压。
  • 2. 本站的文档不包含任何第三方提供的附件图纸等,如果需要附件,请联系上传者。文件的所有权益归上传用户所有。
  • 3. 本站RAR压缩包中若带图纸,网页内容里面会有图纸预览,若没有图纸预览就没有图纸。
  • 4. 未经权益所有人同意不得将文件中的内容挪作商业或盈利用途。
  • 5. 人人文库网仅提供信息存储空间,仅对用户上传内容的表现方式做保护处理,对用户上传分享的文档内容本身不做任何修改或编辑,并不能对任何下载内容负责。
  • 6. 下载文件中如有侵权或不适当内容,请与我们联系,我们立即纠正。
  • 7. 本站不保证下载资源的准确性、安全性和完整性, 同时也不承担用户因使用这些下载资源对自己和他人造成任何形式的伤害或损失。

最新文档

评论

0/150

提交评论