场景分类
该项目主要涉及图像分类。 这里的目标是使用深度神经网络将全球场景分类为六种可能的类别之一。 场景分类的应用范围可以从在智能手机中组织照片,通过旅游业规划促进国家经济增长等方面。
数据集是Kaggle提供的英特尔图像分类。 ( )总共有25,000张图像,其中17,000张被标记为6类,包括建筑物,森林,冰川,山脉,海洋和街道。 使用训练集中的图像对模型进行训练,并预测预测集中的图像类别。
本项目对自训练模型和预训练模型进行了实验,并对所有模型的性能进行了比较和分析。
有关如何运行代码的说明:----
data_prep.py-此文件允许我们加载数据。
vgg16.py vgg19.py res.py inception.py inceptionRes.py-这五个文件使用经过预训练的网络,而不会进行数据扩充。
aug_plots.py-此文件绘制所选图片的增强方式。
2021-06-01 21:14:53
700KB
Python
1