#基于DenseNet网络的猫狗识别
- ###概述
利用DenseNet121神经网络模型,搭配Adam优化器,对猫狗进行二分类识别。
- ###文件描述
CatDog.py:训练模型主程序
ConfusionMatrix.py:输出混淆矩阵、准确率、召回率、精确率、F1 Score
Forecast.py:利用训练好的模型预测猫狗图片
Histogram.py:输出数据集的直方图
ImageEnhancementTest.py:用于测试可视化图像增强的效果
Pretreatment.py:用于对拿到的数据集进行预处理
数据集是由Kaggle提供的猫狗图片,共计25000张,猫与狗各12500张。
作者:Barry