欧美极品高清xxxxhd,国产日产欧美最新,无码AV国产东京热AV无码,国产精品人与动性XXX,国产传媒亚洲综合一区二区,四库影院永久国产精品,毛片免费免费高清视频,福利所导航夜趣136

標(biāo)題: 角蜂鳥python sdk [打印本頁]

作者: aacbwangke    時(shí)間: 2018-11-20 10:06
標(biāo)題: 角蜂鳥python sdk
開發(fā)指南[color=var(--MAIN-ANCHOR-color)][size=0.5em]
hsapi 的包結(jié)構(gòu)如下:
hsapi├── __init__.py├── core│   ├── __init__.py│   ├── base.py│   ├── device.py│   └── graph.py├── high│   ├── __init__.py│   ├── net.py│   └── task.py└── easy    ├── __init__.py    └── prefab.py底層接口[color=var(--MAIN-ANCHOR-color)][size=0.5em]管理設(shè)備 Device實(shí)例[color=var(--MAIN-ANCHOR-color)][size=0.5em]import numpyimport hsapi as hs # 導(dǎo)入 hsapi 模塊, 注意導(dǎo)入路徑# 獲取角蜂鳥設(shè)備device_list = hs.EnumerateDevices() # 獲取所有已連接的角蜂鳥device = hs.Device(device_list[0])# 打開角蜂鳥設(shè)備device.OpenDevice()"""# [可選,根據(jù)業(yè)務(wù)需要]# 讀取角蜂鳥設(shè)備自帶的攝像頭的圖像 image = device.GetImage(zoomMode=True) # zoomMode: True(640x360) False(1920x1080)"""# 操作神經(jīng)網(wǎng)絡(luò)資源、處理數(shù)據(jù)、業(yè)務(wù)邏輯等...# 關(guān)閉設(shè)備,角蜂鳥設(shè)備將會(huì)自動(dòng)重置device.CloseDevice()管理神經(jīng)網(wǎng)絡(luò)資源 Graph實(shí)例[color=var(--MAIN-ANCHOR-color)][size=0.5em]
# 加載Graph文件, 獲取Graph實(shí)例with open('Graph文件路徑', mode='rb') as f:    data = f.read()graph = device.AllocateGraph(data, scale, mean) # scale、mean 圖像預(yù)處理參數(shù)# 神經(jīng)網(wǎng)絡(luò)的圖像輸入,這里有兩種方式 [2選1]# 1. 加載外部圖像,這里以讀取USB攝像頭為例"""import cv2capture = cv2.VideoCapture(0)_, image = capture.read()image = image_preprocess(image) # 把圖像預(yù)處理,處理成神經(jīng)網(wǎng)絡(luò)需要的格式和大小graph.LoadTensor(image.astype(numpy.float16), None) # 加載圖像作為神經(jīng)網(wǎng)絡(luò)的輸入"""# 2. 使用角蜂鳥自帶的攝像頭作為輸入# 使用自帶的攝像頭作為輸入, 返回圖像image = graph.GetImage(True)  # zoomMode: True(640x360) False(1920x1080)# 讀取神經(jīng)網(wǎng)絡(luò)輸出output, _ = graph.GetResult()# 解析神經(jīng)網(wǎng)絡(luò)輸出,實(shí)現(xiàn)業(yè)務(wù)邏輯...# 釋放神經(jīng)網(wǎng)絡(luò)資源graph.DeallocateGraph()簡單的接口調(diào)用流程[color=var(--MAIN-ANCHOR-color)][size=0.5em]# 單角蜂鳥單模型import cv2, numpyimport hsapi as hs # 導(dǎo)入 hsapi 模塊, 注意導(dǎo)入路徑scale = 0.007843 # 圖像預(yù)處理參數(shù)mean = -1.0 # 圖像預(yù)處理參數(shù)device_list = hs.EnumerateDevices() # 獲取所有已連接的角蜂鳥device = hs.Device(device_list[0]) # 獲取Device實(shí)例device.OpenDevice() # 打開角蜂鳥設(shè)備with open('Graph文件路徑', mode='rb') as f:    data = f.read()graph = device.AllocateGraph(data, scale, mean) # 獲取Graph實(shí)例try:    while True:        # 使用自帶攝像頭作為輸入        image = graph.GetImage(True) # 用角蜂鳥設(shè)備圖像作為神經(jīng)網(wǎng)絡(luò)輸入        output, _ = graph.GetResult() # 獲取神經(jīng)網(wǎng)絡(luò)輸出結(jié)果        print(output)        cv2.imshow("horned-sungem", image)        cv2.waitKey(1)finally:    graph.DeallocateGraph() # 釋放神經(jīng)網(wǎng)絡(luò)資源    device.CloseDevice() # 關(guān)閉角蜂鳥設(shè)備高層接口[color=var(--MAIN-ANCHOR-color)][size=0.5em]
[color=var(--MAIN-LINK-color)]Net 是個(gè)抽象類,封裝了部分簡單的設(shè)備操作。
基于該類可實(shí)現(xiàn)一個(gè)自己的神經(jīng)網(wǎng)絡(luò)類,可參考 easy.prefab 模塊中預(yù)置的模型類。
預(yù)置模型 - 人臉檢測[color=var(--MAIN-ANCHOR-color)][size=0.5em]# 單角蜂鳥單模型import cv2, numpy from hsapi import FaceDetector # 導(dǎo)入模塊# 可通過graphPath屬性來指定模型路徑,默認(rèn)路徑在examples/graphs下net = FaceDetector(graphPath="./graph_face_SSD", zoom = True, thresh=0.55) # 創(chuàng)建人臉檢測網(wǎng)絡(luò)try:    while True:        # 使用自帶攝像頭作為輸入        result = net.run(image=None) # 使用角蜂鳥自帶設(shè)備作為輸入,獲取網(wǎng)絡(luò)輸出        image = net.plot(result) # 根據(jù)輸出結(jié)果繪制圖像        cv2.imshow("FaceDetector", image)        cv2.waitKey(1)finally:    net.quit() # 退出神經(jīng)網(wǎng)絡(luò),釋放資源,重置設(shè)備預(yù)置模型 - 人臉檢測 + 物體檢測[color=var(--MAIN-ANCHOR-color)][size=0.5em]# 單角蜂鳥多模型import cv2import hsapi as hsdevice_list = hs.EnumerateDevices() # 獲取所有已連接的角蜂鳥device = hs.Device(device_list[0]) # 獲取Device實(shí)例face_net = hs.FaceDetector(device=device, zoom = True, thresh=0.55) # 創(chuàng)建人臉檢測網(wǎng)絡(luò)obj_net = hs.ObjectDetector(device=device, zoom = True, thresh=0.55) # 創(chuàng)建物體檢測網(wǎng)絡(luò)"""多角蜂鳥多模型: 初始化 Net 時(shí)使用不同的 Device 實(shí)例即可"""capture = cv2.VideoCapture(0) # 獲取USB攝像頭try:    while True:        # 使用USB攝像頭作為輸入        _, image = capture.read() # 讀取USB攝像頭圖像數(shù)據(jù)        face_result = face_net.run(image) # 將圖像輸入到人臉檢測網(wǎng)絡(luò)中        obj_result = obj_net.run(image) # 將圖像輸入到物體檢測網(wǎng)絡(luò)中        image = face_net.overlay(image, face_result[1]) # 根據(jù)人臉檢測網(wǎng)絡(luò)的識(shí)別結(jié)果在圖像上進(jìn)行繪制        image = obj_net.overlay(image, obj_result[1]) # 根據(jù)物體檢測網(wǎng)絡(luò)的識(shí)別結(jié)果在圖像上進(jìn)行繪制        cv2.imshow("Face/Obj Detector", image)        cv2.waitKey(1)finally:    face_net.quit() # 重置設(shè)備,同一個(gè)設(shè)備退出一個(gè)網(wǎng)絡(luò)即可關(guān)于預(yù)處理參數(shù) (scale & mean)[color=var(--MAIN-ANCHOR-color)][size=0.5em]
神經(jīng)網(wǎng)絡(luò)的輸入,根據(jù)網(wǎng)絡(luò)的需要,需要對(duì)圖像進(jìn)行預(yù)處理操作。
當(dāng)使用角蜂鳥自帶攝像頭作為神經(jīng)網(wǎng)路輸入時(shí),目前僅支持對(duì)圖像的RGB通道進(jìn)行統(tǒng)一的預(yù)處理操作。
這里 角蜂鳥設(shè)備內(nèi)部Python API 都使用 乘加 運(yùn)算對(duì)圖像進(jìn)行預(yù)處理操作。
image *= scaleimage += mean
若需要圖像3通道統(tǒng)一處理,或者對(duì)預(yù)處理要求不嚴(yán)格,可對(duì)mean取均值作為參數(shù)輸入。
mean = -1.0 # [-1.0, -1.0, -1.0]scale = 0.007843graph = device.AllocateGraph(data, scale, mean) # scale & meanimage = graph.GetImage(True) # 自帶的設(shè)備圖像已經(jīng)過預(yù)處理后加載到神經(jīng)網(wǎng)絡(luò)中,返回的圖像是未處理的原始圖像output, _ = graph.GetResult() # 讀取神經(jīng)網(wǎng)絡(luò)輸出結(jié)果
若需要灰度圖輸入,或者需要對(duì)不同的通道進(jìn)行不同的預(yù)處理操作,可對(duì)圖像自行處理后加載到角蜂鳥中。
image = preprocess(image) # 自行預(yù)處理graph.LoadTensor(image.astype(numpy.float16), None)API文檔[color=var(--MAIN-ANCHOR-color)][size=0.5em]





歡迎光臨 (http://m.raoushi.com/bbs/) Powered by Discuz! X3.1