Python深度学习基于卷积神经网络实现的猫狗图像分类源码+数据集,猫狗识别源代码
先安装环境---->使用data_classify.py文件进行训练集与测试集分割---->在进行训练即可数据准备:当前数据存放data_name文件夹内文件夹名就是类别名,n个类别就是n个文件夹目录主要结构组成: model_AlexNet.py ---->自己建的AlexNet模型(可选其他模型) model_Vgg16.py ----> pytorch自带更改的模型(可选其他模型) train.py ---->用于训练模型test.py ---->用于测试模型辅助文件: data_classify.py ---->将data_name内的类别分为训练集与测试集。 注意查看代码内容,包含argparse模块清除单通道图像----->数据清洗,处理异常图像旧版数据加载----->用于学习图像数据加载
用户评论