作业3:HotDog-NotHotDog图像分类

采用迁移学习方法,基于VGG16预训练模型,对“Hotdog - Not Hotdog”数据集进行训练。并绘制训练过程中的准确度和验证准确度图表,以展示模型的学习效果。选择一张图片,运用训练好的模型进行预测。

数据集链接:https://www.kaggle.com/datasets/thedatasith/hotdog-nothotdog

参考课件中的12.6“使用VGG16预训练模型进行图像分类”部分:https://www.kaggle.com/code/jeanshendev/neural-network-based-image-classification

作业要求:

1、推荐在kaggle平台上完成作业,打开数据集链接,点击“New Notebook”即可在该数据集上新建Notebook,新建的Notebook包含数据集,可以之间使用。

2、交作业时,将Notebook设置为共享,以kaggle链接的形式交作业。

提示,如何打开GPU计算加速:点击右侧,Session options下方的Accelerator,选择GPU T4 x2,即可开启GPU加速计算。

 

参考答案notebook:https://www.kaggle.com/code/jeanshendev/work3-hotdog-not-hotdog