使用swin_transformer做自己数据集的图像分类

时间:2022-04-30 12:25:12
【文件属性】:

文件名称:使用swin_transformer做自己数据集的图像分类

文件大小:416.23MB

文件格式:ZIP

更新时间:2022-04-30 12:25:12

transformer 分类 文档资料 深度学习 人工智能

使用的数据集共有5种类别,使用Swin-T预训练模型进行训练。


【文件预览】:
swin_transformer
----预训练权重文件下载(70B)
----runs()
--------Apr30_11-56-12_LAPTOP-CDP0EOJT()
--------Apr30_11-58-11_LAPTOP-CDP0EOJT()
--------Apr30_11-55-18_LAPTOP-CDP0EOJT()
----utils.py(7KB)
----class_indices.json(108B)
----data()
--------flower_photos.zip(218.93MB)
----train.py(6KB)
----__pycache__()
--------utils.cpython-37.pyc(5KB)
--------my_dataset.cpython-37.pyc(1KB)
--------model.cpython-37.pyc(20KB)
----predict.py(2KB)
----my_dataset.py(1KB)
----model.py(28KB)
----swin_tiny_patch4_window7_224.pth(109.05MB)
----weights()
--------model-6.pth(105.28MB)

网友评论