whitewaterdeu/image-train-swin-transformer

基于Swin-transformer训练图像分类并部署web端

Image-train-Swin-transformer

基于Swin-transformer训练图像分类并部署web端，解释见博客：https://blog.csdn.net/hi_gril/article/details/118486070?spm=1001.2014.3001.5501
由于Swin-transformer现在只支持训练ImageNet，导致用起来不方便，自己改了下代码，可用于训练自己的数据集

具体包含以下几个步骤：

1.加载预训练权重√

2.图片数据集准备√

3.训练√

4.推理测试√

5.新的数据增强调优

6.部署在web端√
服务器端部署：运行flask_demo，
客户端测试：运行client.py

也可以使用postman测试服务器端