model．fit（train＿generator， epochs＝5， validation＿data＝valid＿generator， callbacks＝［checkpointer］）我們使用一個ModelCheckpoint的回調(diào)來保存驗(yàn)證分?jǐn)?shù)較高的模型。通過測試模型，我們得到的準(zhǔn)確率只有1％左右使用遷移學(xué)習(xí)現(xiàn)在，我們使用遷移學(xué)習(xí)來實(shí)現(xiàn)更高的準(zhǔn)確率。首先我們下載ResNet－50，可以通過運(yùn)行下面的代碼來提取相應(yīng)的訓(xùn)練集、測試和驗(yàn)證集：bottleneck＿features ＝ np．load（＇Data／bottleneck＿features／DogResnet50Data．npz＇）train＿Resnet50 ＝ bottleneck＿features［＇train＇］valid＿Resnet50 ＝ bottleneck＿features［＇valid＇］test＿Resnet50 ＝ bottleneck＿features［＇test＇］我們現(xiàn)在再次定義模型，并對提取的特征使用GlobalAveragePooling2D，它將一組特征平均為一個值。最后，如果驗(yàn)證損失在兩個連續(xù)的epoch內(nèi)沒有增加，我們使用額外的回調(diào)來降低學(xué)習(xí)率；如果驗(yàn)證損失在連續(xù)的5個epoch內(nèi)沒有增加，可以提前停止訓(xùn)練。Resnet50＿model ＝ tf．keras．models．Sequential（）Resnet50＿model．a(chǎn)dd（tf．keras．layers．GlobalAveragePooling2D（input＿shape＝train＿Resnet50．shape［1：］））Resnet50＿model．a(chǎn)dd（tf．keras．layers．Dense（1024， activation＝＇relu＇））Resnet50＿model．a(chǎn)dd（tf．keras．layers．Dense（133， activation＝＇softmax＇））

Resnet50＿model．compile（loss＝＇categorical＿crossentropy＇， optimizer＝＇rmsprop＇， metrics＝［＇accuracy＇］）

checkpointer ＝ tf．keras．callbacks．ModelCheckpoint（filepath＝＇saved＿models／weights＿best＿Resnet50．hdf5＇，                                verbose＝1， save＿best＿only＝True）early＿stopping ＝ tf．keras．callbacks．EarlyStopping（patience＝5， monitor＝＇val＿loss＇）

reduce＿lr ＝ tf．keras．callbacks．ReduceLROnPlateau（patience＝2， monitor＝＇val＿loss＇）Resnet50＿model．fit（train＿Resnet50， train＿targets，           validation＿data＝（valid＿Resnet50， valid＿targets），          epochs＝50， batch＿size＝20， callbacks＝［checkpointer， early＿stopping， reduce＿lr］， verbose＝1）＃＃＃ 訓(xùn)練模型最后在測試集上的準(zhǔn)確率為82．65％，這與我們白手起家訓(xùn)練的模型相比，是一個巨大的進(jìn)步。構(gòu)建web應(yīng)用程序?qū)τ趙eb應(yīng)用程序，我們首先編寫了一個helper函數(shù)，該函數(shù)接受圖像路徑并返回品種。label＿to＿cat字典將每個數(shù)字標(biāo)簽映射到它的狗品種。def predict＿breed（img＿path）：    ＇＇＇預(yù)測給定圖像的品種＇＇＇    ＃ 提取特征    bottleneck＿feature ＝ extract＿Resnet50（path＿to＿tensor（img＿path））    bottleneck＿feature ＝ tf．keras．models．Sequential（［                            tf．keras．layers．GlobalAveragePooling2D（input＿shape＝bottleneck＿feature．shape［1：］）                        ］）．predict（bottleneck＿feature）．reshape（1， 1， 1， 2048）    ＃ 獲得預(yù)測向量    predicted＿vector ＝ Resnet50＿model．predict（bottleneck＿feature）    ＃ 模型預(yù)測的犬種    return label＿to＿cat［np．a(chǎn)rgmax（predicted＿vector）］對于web應(yīng)用程序，我們將使用flaskweb框架來幫助我們用最少的代碼創(chuàng)建web應(yīng)用程序。我們定義一個接受圖像的路由，并用狗的品種呈現(xiàn)一個輸出模板＠app．route（＇／upload＇， methods＝［＇POST＇，＇GET＇］）def upload＿file（）：    if request．method ＝＝ ＇GET＇：        return render＿template（＇index．html＇）    else：        file ＝ request．files［＇image＇］        full＿name ＝ os．path．join（UPLOAD＿FOLDER， file．filename）        file．save（full＿name）        dog＿breed ＝ dog＿breed＿classifier（full＿name）    return render＿template（＇predict．html＇， image＿file＿name ＝ file．filename， label ＝ dog＿breed）predict．html是分別顯示圖像及其犬種的模板。結(jié)論祝賀你！你已經(jīng)成功地實(shí)現(xiàn)了一個狗品種分類器，并且可以準(zhǔn)確地分辨出狗的品種。讓我們總結(jié)一下我們在這里學(xué)到的知識：我們對數(shù)據(jù)集進(jìn)行了分析和預(yù)處理。機(jī)器學(xué)習(xí)算法需要單獨(dú)的訓(xùn)練集、測試集和驗(yàn)證集來進(jìn)行置信預(yù)測。我們從零開始使用CNN，由于未能提取特征，所以表現(xiàn)不佳。然后我們使用了遷移學(xué)習(xí)，準(zhǔn)確度大大提高最后，我們構(gòu)建了一個Flask web應(yīng)用程序來實(shí)現(xiàn)我們的項(xiàng)目封裝我們確實(shí)學(xué)到了很多東西，但你還可以嘗試很多其他的事情。你可以在heroku上部署web應(yīng)用程序，也可以嘗試使用不同的層（如Dropout層）來提高準(zhǔn)確性。參考鏈接：https：／／towardsdatascience．com／dont－know－the－breed－of－your－dog－ml－can－h(huán)elp－6558eb5f7f05

☆ END ☆