Pytorch预训练模型如何设计自己的全连接层?(待完成)
首先构造自己的数据集框架,用ImageFolder完成
参考资料
pytorch 修改预训练模型(全连接层、单个卷积层、多个卷积层)//已实践确实不错!
Pytorch修改ResNet模型全连接层进行直接训练//细节补充
Pytorch修改ResNet模型全连接层进行直接训练 - 嶙羽 - 博客园https://www.cnblogs.com/yqpy/p/11497290.html
pytorch实践(更改resnet50全连接层,重新训练,用于二分类)//非常不错,完整项目
【猫狗数据集】使用预训练的resnet18模型//细细细细
【猫狗数据集】使用预训练的resnet18模型 - 西西嘛呦 - 博客园https://www.cnblogs.com/xiximayou/p/12504579.html