karas + opencv 实现AlexNet神经网络应用于图像分类
一、序言
本项目环境 python 3.8,keras 2.6.0,opencv 4.5.4.58,tensorflow 2.6.0,使用pycharm编译器。 意在使用现有主流框架实现AlexNet经典网络并对猫狗数据集进行训练,实现猫狗的图像分类功能。
目录结构:
data中存有三个文件,image文件夹中猫狗图片各存放三张;model文件夹中有txt文件,存放猫0狗1用于分类;dataset.txt文件中为所有训练照片的索引文字logs文件夹中存放大量训练集训练好的图片,以h5文件格式存储model文件夹中存放AlexNet网络的具体实现train文件夹中为猫狗数据集Test图片文件用于最后训练完毕后的测试功能
二、AlexNet的设计
此文件存放在model文件夹下,使用karas框架构建神经网络
AlexNet网络用于训练数据集,karas框架中的核心数据结构是模型(m
共有 0 条评论