Skip to content

Latest commit

 

History

History
14 lines (13 loc) · 685 Bytes

README.md

File metadata and controls

14 lines (13 loc) · 685 Bytes

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测试服务器端