Skip to content

zmhcasmy/Image-train-Swin-transformer

 
 

Repository files navigation

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

About

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%