简单卷积神经网络图片分类的疑问

2018-10-03 22:24:42 +08:00
 wingspread

放假闲着没事,下载了一段 Python 的简单卷积神经网络,训练了一下,但是识别不准怎么回事? [详细] 基础代码是网上找的猫狗大战的代码,在此基础上自己找了些 ABCD 的答题卡的图片块(不到 200 个),进行训练。然后从训练的图片中找出几个图片进行测试,结果识别的都不正确?有研究 tensorflow 的高手,给指导下吧。

训练的图片太少?

注:相关参数已经修改了,并非直接用下载的代码。两个(卷积层+池化层)+两个全链接层的简单结构。

1431 次点击
所在节点    问与答
6 条回复
Xs0ul
2018-10-03 22:27:20 +08:00
用训练集的都不对的话,那是没训练够吧。先看看你训练时候的准确率,然后要不多训练,要不改参数多训练
wingspread
2018-10-03 22:31:57 +08:00
我试试训练 10000 次的效果吧。原来都 1000 次
squid157
2018-10-03 22:38:53 +08:00
Loss 如果在下降那说明还要继续训练,如果没有就看看哪里有 bug,或者降低 learning rate
kuhung
2018-10-04 00:36:39 +08:00
图片归一化了没?这种情况很可能是格式要求没对上。网上的猫狗有个版本需要对像素值 /255。
shenbo
2018-10-04 11:11:46 +08:00
200 个不够
wingspread
2018-10-05 07:59:04 +08:00
@kuhung 格式统一处理了,统一的大小。代码中的相关参数也修改了。
@shenbo 我再做一些试试。

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/494776

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX