描述

這篇文章包含制作您自己的圖像分析器的說(shuō)明。

硬件要求

• 樹(shù)莓派
• 樹(shù)莓派相機(jī)
• 字符型液晶顯示器
• 跳線(xiàn)
• 面包板
• 電位計(jì)（如果您的 LCD 支持對(duì)比度調(diào)整）
• 按鈕

請(qǐng)檢查下面的示意圖以獲取接線(xiàn)圖。

軟件要求

我們將在按下按鈕時(shí)拍攝照片，然后將該圖像上傳到 Dropbox。然后，我們將從 Dropbox 獲取可共享鏈接，并使用該照片鏈接調(diào)用 Microsoft Computer Vision API。API 調(diào)用返回的結(jié)果將顯示在字符 LCD 上。

安裝依賴(lài)項(xiàng)

Adafruit Python CharLCD 庫(kù)

從以下 Github 鏈接獲取 LCD 庫(kù)。

https://github.com/adafruit/Adafruit_Python_CharLCD  

CD 到下載的文件夾并運(yùn)行以下命令。

sudo python3 setup.py install   

要求

sudo pip3 install requests   

投遞箱

登錄您的保管箱帳戶(hù)并創(chuàng)建一個(gè)應(yīng)用程序來(lái)存儲(chǔ)捕獲的圖像。您還需要一個(gè)訪(fǎng)問(wèn)令牌。以下鏈接可能會(huì)有所幫助。

通過(guò)鍵入以下命令安裝 dropbox python 客戶(hù)端庫(kù)

sudo pip3 install dropbox   

獲取計(jì)算機(jī)視覺(jué) API（訂閱）密鑰以進(jìn)行 API 調(diào)用

您可以通過(guò)訪(fǎng)問(wèn)以下鏈接獲取免費(fèi)訂閱密鑰。在撰寫(xiě)本文時(shí)，Microsoft 允許每月進(jìn)行 5000 次免費(fèi)交易

Python 3 代碼

導(dǎo)入依賴(lài)項(xiàng)

import RPi.GPIO as GPIO  
import picamera  
import Adafruit_CharLCD as LCD  
import os.path  
import time  
import requests  
import dropbox  
import json   

調(diào)用 API 的變量

使用您之前獲得的密鑰更新 subscription_key 變量。

subscription_key = 'YOUR_Subscription_KEY'  
# URL will be mentioned when you obtain your  
url = "https://westcentralus.api.cognitive.microsoft.com/vision/v1.0/analyze"   

Dropbox 的訪(fǎng)問(wèn)令牌

access_token = 'YOUR_ACCESS_TOKEN'   

樹(shù)莓派開(kāi)發(fā)板

GPIO.setmode(GPIO.BCM)   

按鈕設(shè)置

btn = 16  
GPIO.setup(btn, GPIO.IN, pull_up_down=GPIO.PUD_UP)   

驅(qū)動(dòng) LCD 的變量

對(duì)應(yīng)的 Raspberry Pi GPIO 引腳用于 LCD 上的寄存器選擇、指令和數(shù)據(jù)寄存器。稍后我們將使用這些變量來(lái)初始化 LCD。

lcd_rs = 4  
lcd_en = 17  
lcd_d4 = 25  
lcd_d5 = 24  
lcd_d6 = 23  
lcd_d7 = 18   

LCD 的尺寸，如果您有 16x2 的字符 LCD，則將 lcd_columns 變量修改為 16，將 lcd_rows 修改為 2。由于我使用的是 20x4 LCD，因此我將使用以下值。

lcd_columns = 20  
lcd_rows = 4   

初始化

通過(guò)調(diào)用 Dropbox、Adafruit 和 picamera 庫(kù)提供的構(gòu)造方法來(lái)初始化 Dropbox、LCD 和相機(jī)。如果您的 LCD 支持背光，那么您可以傳入一個(gè)附加參數(shù) backlight = 1 或 backlight = 0 來(lái)打開(kāi)或關(guān)閉它。

# Initilaize Dropbox  
dbx = dropbox.Dropbox(access_token)  
# Initialize the LCD using the pins above by calling the construct method of Adafruit LCD library  
lcd = LCD.Adafruit_CharLCD(lcd_rs, lcd_en, lcd_d4, lcd_d5, lcd_d6, lcd_d7, lcd_columns, lcd_rows)  
# Initialize the camera  
cam = picamera.PiCamera()   

主函數(shù)

main() 函數(shù)是程序啟動(dòng)時(shí)第一個(gè)被調(diào)用的函數(shù)。它等待按鈕按下，當(dāng)按下按鈕時(shí)，它調(diào)用其他函數(shù)來(lái)捕獲圖像，將其上傳到保管箱，獲取上傳的圖像鏈接，然后最后調(diào)用 predict_contents() 函數(shù)，該函數(shù)調(diào)用 Vision API 并在 LCD 上顯示結(jié)果。當(dāng)用戶(hù)再次按下按鈕時(shí)，在最后一步調(diào)用 main() 函數(shù)以執(zhí)行相同的步驟。

def main():  
  while True:  
    cam.capture('image.jpg')  
    display_message('Uploading..')  
    upload_image()  
    display_message('Generating Image', 'link..')  
    image_url = get_image_url()  
    display_message('Analysing..')  
    predict_contents(image_url)  
    main()   

下面將更詳細(xì)地描述此處使用的每個(gè)函數(shù)。

display_message() 函數(shù)

display_function() 函數(shù)利用 Adafruit LCD 庫(kù)提供的 message() 函數(shù)。調(diào)用它時(shí)最多可以傳入四個(gè)字符串，它會(huì)在新行上顯示每個(gè)字符串。但是，如果您只傳入 line1 參數(shù)，它將在多行中顯示消息。lcd.clear()將清除 LCD 上的任何現(xiàn)有文本 lcd.home() 并將光標(biāo)移至第 1 行和第 1 列。

def display_message(line1 = '', line2 = '', line3 = '', line4 = ''):  
  lcd.clear() # Clear the display  
  lcd.home() # Bring cursor to row 1 and column 1  
  if(line1 != '' and line2 == '' and line3 == '' and line4 == ''):  
    lcd.message(line1)  
  else:  
    lcd.message(line1 + '\n' + line2 + '\n' + line3 + '\n' + line4)  

upload_image() 函數(shù)

這段代碼在命名的工作目錄中獲取先前捕獲的圖像image.jpg并將其上傳到 dropbox。

def upload_image():  
  remote_file = '/image.jpg'  
  f = open(local_file, 'rb')  
  dbx.files_upload(f.read(), remote_file, mode=dropbox.files.WriteMode.overwrite)   

get_image_url() 函數(shù)

此函數(shù)向 Dropbox 發(fā)出發(fā)布請(qǐng)求，以獲取我們?cè)谏弦徊缴蟼鞯膱D像的可共享鏈接。然后將結(jié)果轉(zhuǎn)換為 JSON。image_url.replace('www', 'dl')以便可以下載。此步驟是必要的，否則圖像會(huì)在 Dropbox Web 應(yīng)用程序中打開(kāi)。此處使用之前從 Dropbox 獲取的訪(fǎng)問(wèn)令牌對(duì)請(qǐng)求進(jìn)行身份驗(yàn)證。

def get_image_url():  
  headers = {  
    'Authorization': 'Bearer ' + access_token,  
    'Content-Type': 'application/json'  
  }  
  data = {  
    "path": "/image.jpg"  
  }  
  r = requests.post('https://api.dropboxapi.com/2/sharing/create_shared_link', headers=headers, data=json.dumps(data))  
  image_url = r.json()['url']  
  image_url = image_url.replace('www', 'dl')  
  return image_url   

predict_contents(image_url) 函數(shù)

該函數(shù)將之前獲取的圖片鏈接作為參數(shù)傳入計(jì)算機(jī)視覺(jué)API。訂閱密鑰用于對(duì)請(qǐng)求進(jìn)行身份驗(yàn)證，visualFeatures參數(shù)包含有關(guān)我們要在圖像中分析哪些特征的詳細(xì)信息。Vision API 應(yīng)用機(jī)器學(xué)習(xí)來(lái)識(shí)別圖像并返回結(jié)果。標(biāo)題最好用一個(gè)句子概括圖像，標(biāo)簽包含有關(guān)圖像中有哪些特征的信息。前 8 個(gè)最有信心的標(biāo)簽顯示在 LCD 上的標(biāo)題后。

def predict_contents(image_url):  
  params   = {'visualFeatures': 'Description'}  
  data     = {'url': image_url}  
  response = requests.post(url, headers=headers, params=params, json=data)  
  response.raise_for_status()  
  image_desc = response.json()['description']['captions'][0]['text']  
  image_tags = response.json()['description']['tags']  
  display_message(image_desc)  
  time.sleep(5)  
  display_message('{}, {},'.format(image_tags[0], image_tags[1]), '{}, {},'.format(image_tags[2], image_tags[3]), '{}, {},'.format(image_tags[4], image_tags[5]), '{}, {}'.format(image_tags[6], image_tags[7]),)  
  time.sleep(10)   

我們應(yīng)用程序的起點(diǎn)

這是我們調(diào)用main()函數(shù)并在程序停止時(shí)顯示“程序停止”的地方。

try:  
  main()  
finally:  
  display_message('Program Stopped.')   

運(yùn)行應(yīng)用程序

最后，我們到了啟動(dòng)程序并享受拍照的樂(lè)趣。https://github.com/dsdilpreet/blog/blob/master/recognize-things-with-raspberry-pi-vision

從 github 下載文件，打開(kāi)終端并 cd 進(jìn)入下載文件所在的文件夾并運(yùn)行以下命令。

python3 app.py   

帶有示例圖片和結(jié)果鏈接的原始帖子

?