遷移學習:keras + vgg16 + cifar10 實現圖像識別

框架:keras 數據集:CIFAR10 模型:vgg16 注:vgg16模型的輸入圖像尺寸至少爲 48*48 思路:去掉vgg16的頂層,保留其餘的網絡結構與訓練好的權重。然後添加模型結構,進而訓練CIFAR10。 1.模型結構 2.具體代碼以及註釋 ①訓練代碼 #-*- coding: utf-8 -*- #遷移學習,vgg16+cifar10 from keras.applications.
相關文章
相關標籤/搜索