如何用keras实现图片多分类?

发布于 2019-06-20 10:13:19

问题描述:一个图片10分类问题,图片大小统一224224,单通道。训练集每类1000张,分10类,每类放在一个文件夹下,共100010张;验证集每类200张,分10类,每类放在一个文件夹下,总共20010张;测试集每类200张,分10类,每类放在一个文件夹下总共20010张。现在想在windows下用keras框架(后台是theano)来对图片进行分类测试。

官方教程里的二分类例子:

http://blog.csdn.net/yichenmoyan/article/details/51908840(翻译)

例子源码地址:

https://gist.github.com/fchollet

classifier_from_little_data_script_1.py是第一步简单模型源码

classifier_from_little_data_script_2.py及

classifier_from_little_data_script_3.py是第二步在vgg-16网络上进行修改

我希望在这个例子的基础上修改成多分类的,请各位帮下忙!!!

查看更多

关注者
0
被浏览
252
1 回答
哈尼鹿哈
哈尼鹿哈 2019-06-20
一点点成长也是改变,加油!

0.注意一下y应该是一个向量,(这里是10维向量)可以用np_utils.to_categorical把类别对应的数字转换乘向量。(from keras.utils import np_utils)1.模型最后两行改为, model.add(Dense(10)),model.add(Activation('softmax'))2.model编译的loss改为categorical_crossentropy

撰写答案

请登录后再发布答案 点击登录

发布
问题

分享
好友

手机
浏览

扫码手机浏览