【PyTorch深度学习项目实战100例】—— 基于ResNet50实现多目标美味蛋糕图像分类 | 第51例

前言

大家好，我是阿光。

本专栏整理了《PyTorch深度学习项目实战100例》，内包含了各种不同的深度学习项目，包含项目原理以及源码，每一个项目实例都附带有完整的代码+数据集。

正在更新中~ ✨

🚨 我的项目环境：

• 平台：Windows10
• 语言环境：python3.7
• 编译器：PyCharm
• PyTorch版本：1.8.1

💥 项目专栏：【PyTorch深度学习项目实战100例】

---

一、基于ResNet50实现多目标美味蛋糕图像分类

本项目是一个使用ResNet50实现图像分类的任务，但是不同于传统分类，本项目是一个多目标分类任务，传统分类任务需要判别一张图片的单一类别，例如判别是猫或是狗，但是如果我们有一种需求就是如果一张图片中既有狗又有猫，就需要将两个类别同时判为1，本项目是判断一个蛋糕中是否同时有奶油、水果、撒料，是一个多目标分类任务。

二、数据集介绍

数据说明：