采用深度学习方法进行图像识别,数据集为kaggle数据集中的猫与狗数据集。 由于自己的硬件条件,目前只是提取了2000张图像作为训练集,1000张作为验证集,1000张作为测试集,如果有朋友想用全部数据集作为训练,可以更改源码,自己训练。目前采用猫与狗数据分类识别率到达97%识别率,采用了数据增强,卷积神经网络作为分类。 同时脚本中含有基于keras框架的一些简单分类脚本,包含回归、多分类等。
kmustcti404 / kaggle-cats-and-dogs Goto Github PK
View Code? Open in Web Editor NEWThis project forked from zw76859420/kaggle-cats-and-dogs
采用深度学习方法进行图像识别,数据集为kaggle数据集中的猫与狗数据集。