源码地址:https://github.com/SwinTransformer/Swin-Transformer-Object-Detection
参考目录:
https://www.bilibili.com/video/BV1eu41197qK/?spm_id_from=333.788
https://www.bilibili.com/video/BV1Y3411676J?spm_id_from=333.337.search-card.all.click
简单记录下,安装、跑通、GPU训练swin-t源码的步骤。
环境配置
- 首先从官网获得源码:https://github.com/SwinTransformer/Swin-Transformer-Object-Detection
git clone https://github.com/SwinTransformer/Swin-Transformer-Object-Detection.git
-
新建conda环境、安装pytorch 我用的是python==3.6.13和pytorch==1.7.1和 torchvision==0.8.2
安装pytorch的教程很多,这里就不赘述了,GPU版本注意和cuda驱动的匹配。验证方法我这边放一下:
$ python
进入python命令行
import torch
没有出现错误就ok了
如果是GPU版本还需要输入torch.cuda.is_available()
出现True,就是成功了。&