- 博客(6)
- 收藏
- 关注
RSS订阅原创 用训练得到的pytorch图像分类模型，识别图像、视频、摄像头画面
【子豪兄带你两天搞定AI毕业设计】使用上一讲Pytorch迁移学习训练得到的30类水果图像分类模型，对新图像文件、新视频文件、摄像头实时画面，运行图像分类预测。代码教程：https://github.com/TommyZihao/Train_Custom_Dataset代码运行云GPU环境：https://featurize.cn/?s=d7ce99f842414bfcaea5662a97581bd1
2023-01-26 23:10:12
730
转载 Pytorch迁移学习训练自己的图像分类模型
使用深度学习框架Pytorch，在ImageNet预训练图像分类模型（例如Resnet18）基础上，对自己图像分类数据集进行迁移学习（transfer learning）微调（fine-tuning）训练，得到自己的图像分类模型。在训练过程中，记录训练集和测试集的损失函数、准确率、Precision、Recall、f1-score等评估指标，使用wandb可视化面板监控。为后续的新图像预测、测试集评估、可解释性分析、模型部署，奠定算法基础。
2023-01-24 23:45:36
333
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人