transformer训练自己的图片数据集

一、准备工作

1.1 源码地址：

​​https://github.com/SwinTransformer/Swin-Transformer-Object-Detection​​

Pytorch环境

已经安装pytorch gpu版本

安装mmcv

切记版本

git clone -b v1.3.1 https://github.com/open-mmlab/mmcv.git

cd mmcv
pip install -r requirements.txt

进入anaconda powershell prompt环境执行操作，

但是anaconda 找不到anaconda powershell prompt

%windir%\System32\WindowsPowerShell\v1.0\powershell.exe -ExecutionPolicy ByPass -NoExit -Command "& 'D:\Anaconda3\shell\condabin\conda-hook.ps1' ; conda activate 'D:\Anaconda3' "

mmcv跟着视频教程，没有安装成功

pip install mmcv-full==1.3.8 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.0/index.html

这个命令也不行

参考：

​​https://www.bilibili.com/video/BV1QY4y1i7on/​​

视频教程：Swin Transformer实战目标检测：训练自己的数据集

VOC数据集转coco数据集