logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

Pytorch迁移学习训练自己的图像分类模型

使用深度学习框架Pytorch,在ImageNet预训练图像分类模型(例如Resnet18)基础上,对自己图像分类数据集进行迁移学习(transfer learning)微调(fine-tuning)训练,得到自己的图像分类模型。在训练过程中,记录训练集和测试集的损失函数、准确率、Precision、Recall、f1-score等评估指标,使用wandb可视化面板监控。为后续的新图像预测、测试集

#pytorch#迁移学习#分类
到底了