開啟圖像分類的實踐之路

第一步　創建模型

說時遲，那時快，哪吒已經摩拳擦掌地準備大顯身手了！首先，第一個任務是確定模型類型，配置模型基本信息，并記錄希望模型實現的功能。

1）打開EasyDL平臺主頁，網址為https://ai.baidu.com/easydl/。

點擊圖2-7中的【立即使用】按鈕，會顯示如圖2-8所示的【選擇模型類型】選擇框，選擇模型類型為【圖像分類】，進入圖2-9所示的操作臺界面。

圖2-7　EasyDL平臺主頁

圖2-8　選擇模型類型

圖2-9　操作臺界面

2）創建模型。點擊操作臺頁面中的【創建模型】按鈕，顯示如圖2-10所示的界面。在該界面中填寫模型名稱為“蝦蟹分類”，模型歸屬選擇“個人”，填寫聯系方式、功能描述等信息，點擊【完成】按鈕，完成模型的創建。

圖2-10　創建模型

3）模型創建成功后，就可以在【我的模型】中看到剛剛創建的“蝦蟹分類”模型，如圖2-11所示。

圖2-11　模型列表

第二步　準備數據

這個階段的主要工作是根據具體圖像分類的任務準備相應的數據集，并把數據集上傳到EasyDL平臺，用來訓練模型。

（1）準備數據集

首先，準備用于訓練模型的圖像數據。對于蝦蟹分類任務，我們準備了龍蝦和螃蟹兩種類型的圖像。圖片格式均為jpg，除此之外還支持png、bmp、jpeg格式的圖片。

然后，將準備好的圖像數據按照分類存放在不同的文件夾里，文件夾名稱即為圖像對應的類別標簽（longxia、pangxie）。此處要注意，圖像類別名即文件夾名稱只能包含字母、數字、下劃線，不支持中文命名。

最后，將所有文件夾壓縮，命名為xiaxie.zip，壓縮包的結構示意圖如圖2-12所示。

圖2-12　壓縮包的結構示意圖

（2）上傳數據集

點擊圖2-13所示的【數據總覽】中的【創建數據集】按鈕，進行數據集的創建。如圖2-14所示，填寫數據集名稱及標注類型，點擊【完成】按鈕，然后在“數據總覽”頁面中點擊【導入】，數據標注狀態選擇【有標注信息】，導入方式選擇【本地導入】并上傳壓縮包xiaxie.zip，上傳完成后點擊【確認并返回】按鈕，如圖2-15和圖2-16所示。

圖2-13　創建數據集

圖2-14　填寫數據集信息

圖2-15　數據集列表

圖2-16　上傳數據集

（3）查看數據集

上傳成功后，可以在【數據總覽】中看到數據集正在導入，如圖2-17所示。數據集上傳后，需要一段處理時間，大約幾分鐘后就可以看到數據上傳的結果了，如圖2-18所示。

圖2-17　數據集導入中

圖2-18　數據集導入成功

點擊【查看與標注】，可以看到圖像被分為longxia和pangxie兩個標簽，如圖2-19所示。

第三步　訓練模型并校驗結果

經過前兩步，已經創建好了一個圖像分類模型，并且創建了數據集，本步驟的主要任務是用上傳的數據訓練模型，并且在模型訓練完成后，在線校驗模型的效果。

圖2-19　數據標注

（1）訓練模型

經過第二步上傳數據成功后，在【我的模型】界面中，點擊“蝦蟹分類”模型項目的【訓練】按鈕，在【訓練模型】界面中，選擇之前創建的圖像分類模型，添加分類數據集，就可以開始訓練模型。訓練時間與數據量有關。這個過程如圖2-20、圖2-21、圖2-22和圖2-23所示。

圖2-20　點擊【訓練】按鈕

圖2-21　添加分類數據集

圖2-22　訓練模型

圖2-23　模型訓練中

（2）查看模型效果

模型訓練完成后，在【我的模型】列表中可以看到模型效果，如圖2-24所示。點擊圖2-24中的【完整評估結果】，可以看到模型訓練的整體情況說明，結果顯示該模型的訓練效果是比較優異的，如圖2-25所示。

圖2-24　模型訓練結果

圖2-25　模型整體評估

（3）校驗模型

點擊圖2-26中的【啟動模型校驗服務】按鈕，大約等待5分鐘后，進入【校驗模型】界面，如圖2-27所示。

圖2-26　模型校驗

圖2-27　校驗模型界面

然后，準備一條圖像數據，在【校驗模型】界面中點擊“點擊添加圖片”按鈕添加圖像。

最后，使用訓練好的模型對上傳的圖像進行預測，如圖2-28所示，成功識別出圖像中的物體為螃蟹。

圖2-28　校驗結果

只見不出半小時，哪吒就把敖丙的三千蝦兵蟹將全部辨認清楚了，敖丙一行人驚得目瞪口呆。哪吒昂首闊步地走到敖丙面前，義正詞嚴地說道：“小爺我做到了！回去告訴你父王趕緊給陳塘關施雨，以后休得再興風作浪！”

“啥？你說啥？”敖丙裝傻充愣地環顧四周說道：“你們剛才聽到我說什么了嗎？”“沒有，沒有！”蝦兵蟹將的嘲笑聲不絕于耳。

這可把哪吒氣得肝膽欲碎、發指眥裂，他咬牙切齒道：“敖丙，你這無恥小人，不守諾言，為非作歹，無可救藥！小爺我今日要給你點顏色看看！”

話音未落，只見敖丙手持方天畫戟向哪吒刺來，哪吒縱身一閃，讓敖丙撲了個空。敖丙惱羞成怒，示意蝦兵蟹將沖上去圍攻。哪吒忍無可忍，一把甩出混天綾把三太子牢牢捆住，乾坤圈一出，直接擊中了敖丙的頭部，瞬間將他打回原形。這一戰哪吒鋒芒畢露，蝦兵蟹將見三太子被打死，嚇得魂不附體，連滾帶爬地鉆回了水里。哪吒見這家伙元神竟是一條通體晶瑩剔透的小白龍，想起父親常年征戰，腰帶都磨損了，龍筋可是做腰帶上好的材料，于是“唰”的一聲抽出龍筋，高高興興地揚長而去（見圖2-29）。

圖2-29　哪吒怒抽龍筋