最近需要使用图像分类算法来实现一些功能,找来找去发现北村南大佬已经使用COIL20数据集分别在AlexNet、VGG16、GoogleNet等共5个模型上进行了实验,于是决定在windows上复现一下其项目,然后保存训练好的模型,并调用训练好的模型进行图像分类,现记录一下过程。
北村南大佬的原文链接:
【A图像分类实战】2024 Pytorch搭建AlexNet、VGG16、GoogleNet等共5个模型实现COIL20数据集图像20分类完整项目(项目已开源)
最近需要使用图像分类算法来实现一些功能,找来找去发现北村南大佬已经使用COIL20数据集分别在AlexNet、VGG16、GoogleNet等共5个模型上进行了实验,于是决定在windows上复现一下其项目,然后保存训练好的模型,并调用训练好的模型进行图像分类,现记录一下过程。
北村南大佬的原文链接:
【A图像分类实战】2024 Pytorch搭建AlexNet、VGG16、GoogleNet等共5个模型实现COIL20数据集图像20分类完整项目(项目已开源)