README.md

# 数据集
imagenet2012 
```
apt-get install git-lfs

git lfs clone http://113.200.138.88:18080/aidatasets/project-dependency/imagenet-2012.git
```


# 下载模型


```
模型：swinv2-large-w24_in21k-pre_3rdparty_in1k-384px
wget https://github.com/open-mmlab/mmpretrain.git
```

# 环境搭建

```
cd mmpretrain
pip install openmim
mim install -e .  -i https://pypi.tuna.tsinghua.edu.cn/simple 
```


# 训练

## 单卡执行
```
# 修改

vim configs/_base_/models/swin_transformer_v2/large_384.py
vim configs/_base_/datasets/imagenet_bs64_swin_384.py
vim configs/_base_/schedules/imagenet_bs1024_adamw_swin.py
vim configs/_base_/default_runtime.py

python tools/train.py  configs/swin_transformer_v2/swinv2-large-w24_in21k-pre_16xb64_in1k-384px.py  --amp

```
## 多卡执行
```
bash ./tools/dist_train.sh  configs/swin_transformer_v2/swinv2-large-w24_in21k-pre_16xb64_in1k-384px.py  8 --amp
```


# 推理

```
python tools/test.py   configs/swin_transformer_v2/swinv2-large-w24_in21k-pre_16xb64_in1k-384px.py   
```




# 参考文档


https://github.com/open-mmlab/mmpretrain.git  
https://github.com/open-mmlab/mmpretrain/tree/main/configs/swin_transformer_v2