Python識別花卉種類，并自動整理分類！
今天我們來鑒定一下網(wǎng)絡熱門植物！最近春天很多花都開了，我正好趁著假期到戶外踏青并拍攝了不少花卉的照片。
由于對很多花不是特別熟悉，所以我們需要借助軟件來識別究竟是什么花的種類。
市面上的識花軟件有很多，比如花伴侶、形色、百度等等，我測試后發(fā)現(xiàn)百度的識別效果最為優(yōu)秀。
于是我就有了一個想法，能不能批量調(diào)用百度的接口，對花卉照片進行識別并分類呢？（完整代碼見文末）
百度圖像識別
百度的圖像識別接口，可以精準識別超過十萬種物體和場景，包含10余項高精度的識圖能力并提供相應的API服務。
https://cloud.baidu.com/product/imagerecognition
我們按照步驟創(chuàng)建新應用，并得到專屬自己的API Key和Secret Key，具體如下圖所示。
圖像識別提供一個組合API接口，支持多種分類識別服務的靈活組合調(diào)用，這里只需要調(diào)用植物識別就可以滿足需求。
如何Python調(diào)用百度圖像識別API接口？
第一步，調(diào)用鑒權接口獲取token。
API_Key?=?'**********'
Secret_Key?=?'**********'
def?get_access_token(API_Key,Secret_Key):?
????host?=?'**********'
????response?=?requests.get(host)
????return?response.json()['access_token']
access_token?=?get_access_token(API_Key,Secret_Key)
第二步，識別圖像種類
在交互式環(huán)境中輸入如下命令：
import?requests
import?base64
request_url?=?'**********'
#?二進制方式打開圖片文件
f?=?open(r'D:\下載\QQ截圖20220407203203.png',?'rb')
img?=?base64.b64encode(f.read())
params?=?{"image":img}
request_url?=?request_url?+?"?access_token="?+?access_token
headers?=?{'content-type':?'application/x-www-form-urlencoded'}
response?=?requests.post(request_url,?data=params,?headers=headers)
if?response:
????print?(response.json()['result'][0]['name'])
輸出：
櫻花
調(diào)用百度圖像識別接口的返回參數(shù)如下圖所示，對我們來說，只需要其中的name（植物名稱）參數(shù)。
讀取照片文件
我將拍攝的照片存放在D:\下載\花卉合集路徑下，所以需要使用os模塊進行讀取文件列表，方便進行后續(xù)的批量操作。
在交互式環(huán)境中輸入如下命令：
import?os
path?=?"D:\下載\花卉合集"
filenames?=?os.listdir(path)
filenames
輸出：
['QQ截圖20220405223301.png',
'QQ截圖20220405223320.png',
......
'微信圖片_20220405225020.jpg',
'微信圖片_20220405225023.jpg']
os模塊中的listdir()?法，接收?個路徑參數(shù)path，返回的是該路徑下所有?件的?件名組成的列表。這樣，我們就獲取了該路徑下所有的花卉圖片文件名，如下圖所示。
整理分類照片
接著，我們便可以使用for循環(huán)語句，依次對花卉照片進行圖像識別，并按照識別出的名稱進行分類整理到對應的文件夾中。
在交互式環(huán)境中輸入如下命令：
for?i?in?filenames:
????flower_name?=?get_fname(i)
????file_path?=?os.path.join(path,i)
????folder_path?=?os.path.join(path,flower_name)
????if?not?os.path.exists(folder_path):?
????????os.mkdir(folder_path)
????shutil.move(file_path,folder_path)
其中get_fname()函數(shù)，是我們將前文中百度圖像識別代碼封裝為一個自定義函數(shù)，此處調(diào)用即可返回得到照片對應的花卉名稱flower_name。
后續(xù)的代碼與之前分享過的自動分類整理文件幾乎一致，即if判斷是否已經(jīng)存在對應花卉名稱的文件夾，若不存在則創(chuàng)建。最后，調(diào)用shutil模塊移動花卉照片至對應文件夾。
具體執(zhí)行效果，如下方動圖所示。
大家如果本文涉及的代碼感興趣，在公眾號后臺回復“賞花”即可獲取對應【圖片+完整代碼】文件。
- end -