- Notifications
You must be signed in to change notification settings - Fork0
ddm3114/General-Image-Classification-Framework
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
目前框架内集成了:
在CUB-200-2011数据集上从零开始训练新的输出层,并对其余参数使用较小的学习率进行微调。
使用框架resnet18
resnet34和swin transfomer
自定义网络(两个baseblock作为主干)
- model.py
- get_model.py
- config.json
- dataset.py
- get_dataloader.py
- config.json
- get_optim.py
- 自定义网络的全部参数或者线性分类头(允许分别训练)
- 预训练网络中的全部参数和线性分类头(允许分别训练)
对框架有问题可联系:学邮:21307140101@m.fudan.edu.cn 私人邮箱:latzbnzl@gmail.com
About
感谢复旦大学,不同老师布置的相同的图像分类作业。让我总结这么一套框架
Resources
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Releases
No releases published
Packages0
No packages published
Contributors2
Uh oh!
There was an error while loading.Please reload this page.