之前的文章介紹了使用Keras解決二分類問題。那么對(duì)于多分類問題該怎么解決?本文介紹利用深度學(xué)習(xí)----Keras進(jìn)行多分類。
1. 準(zhǔn)備數(shù)據(jù)集
為了演示,本次選用了博文keras系列︱圖像多分類訓(xùn)練與利用bottleneck features進(jìn)行微調(diào)(三)中提到的數(shù)據(jù)集,原始的數(shù)據(jù)集將所有類別的train照片放到train文件夾中,所有的test照片放在test文件夾中,而用不同數(shù)字開頭來表示不同類別,比如以3開頭的照片就是bus類等。首先將這些不同類別的照片放在不同的文件夾中,最終的train文件夾有5個(gè)子文件夾,每個(gè)子文件夾中有80張圖片,最終的test文件夾中有5個(gè)子文件夾,每個(gè)子文件夾中有20張圖片??偣仓挥?00張圖片。文章來源:http://www.zghlxwxcb.cn/news/detail-732858.html
在代碼上,需要用ImageDataGenerator來做數(shù)據(jù)增強(qiáng),并且用flow_from_directory來從文件夾中產(chǎn)生數(shù)據(jù)流。
代碼和二分類的文章基本相同,唯一的不同之處是要設(shè)置class_mode='categorical',而不是原來二分類問題的class_mode='binary'
2. 模型的構(gòu)建和訓(xùn)練
基本和二分類一樣,如下為模型的構(gòu)建部分:</文章來源地址http://www.zghlxwxcb.cn/news/detail-732858.html
到了這里,關(guān)于深度學(xué)習(xí)使用Keras進(jìn)行多分類的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!