引言
OpenVINO2022版本發布之后,對之前舊版本的功能做明確的劃分,其中仍然通過exe方式安裝程序的是runtime推理包,支持ONNX、IR、PADDLE等模型讀取與推理。但是模型優化轉換、模型庫下載功能等其他功能被分在一個叫Dev Tool的部分。該部分可以通過pip方式直接安裝,然后通過命令行直接直線,完成模型的轉換,下載等操作,跟之前版本易用性有較大提升!做個對比如下:
Dev Tools安裝與使用
Dev Tools安裝非常方便,直接通過官方腳本命令行選擇安裝即可,唯一需要注意的是選擇模型框架支持,我選擇了ONNX/Pytorch格式轉換支持,安裝的命令行如下:
pip install openvino-dev[onnx,pytorch]==2022.1.0
執行完這條命令行就算是安裝好了,只要網絡不掛就可以安裝成功!
安裝完成,轉換一個ONNX格式的模型為IR格式(xml/bin)文件,以Pytorch的ResNet18為例,先轉換為ONNX,代碼如下:model = models.resnet18(pretrained=True)
model.eval()
model.cpu()
dummy_input1 = torch.randn(1, 3, 224, 224)
torch.onnx.export(model, (dummy_input1), "resnet_model.onnx", verbose=True)
然后直接運行命令行就可以轉換IR格式文件,截圖如下:
模型下載
安裝完成Dev Tools之后,下載模型,只要執行命令行即可:舉例如下:
omz_downloader --name person-detection-0200
表示下載模型person-detection-0200是一個輕量化的人臉檢測模型。omz_downloader支持的參數:--all表示下載全部模型,建議別這么干!
--name 下載一個或者多個指定名稱的模型,推薦這么干!
--precisions 表示下載的模型精度參數,支持FP32/FP18/INT8
對比之前Python版本的SDK,好用了不少,最明顯的感受就是不用讀輸入輸出,然后一堆設置了,對單個輸入輸出的網絡,調用就特別的簡潔更方便!開發者更加容易上手!上面我已經成功轉換一個pytorch圖像分類模型為IR格式,現在就可以使用它,基于OpenVINO2022版本最新Python SDK部署調用,實現代碼如下:
#加載標簽數據
withopen('imagenet_classes.txt')asf:
labels=[line.strip()forlineinf.readlines()]
defresnet_demo():
ie=Core()
#model=ie.read_model(model="resnet_model.onnx")
model=ie.read_model(model="resnet_model.xml")
compiled_model=ie.compile_model(model=model,device_name="CPU")
output_layer=compiled_model.output(0)
means=np.zeros((224,224,3),dtype=np.float32)
means[:,:]=(0.485,0.456,0.406)
dev=np.zeros((224,224,3),dtype=np.float32)
dev[:,:]=(0.229,0.224,0.225)
image=cv.imread("D:/images/space_shuttle.jpg")
rgb=cv.cvtColor(image,code=cv.COLOR_BGR2RGB)
#resizetoMobileNetimageshape
input_image=cv.resize(src=rgb,dsize=(224,224))
blob_img=np.float32(input_image)/255.0
input_x=(blob_img-means)/dev
input_x=input_x.transpose((2,0,1))
input_x=np.expand_dims(input_x,0)
print(input_x.shape)
result_infer=compiled_model([input_x])[output_layer]
result_index=np.argmax(result_infer)
cv.putText(image,labels[result_index],(20,50),cv.FONT_HERSHEY_SIMPLEX,1.0,(0,0,255),2)
cv.imshow("OpenVINO2022+PythorchResNet18",image)
cv.waitKey(0)
cv.destroyAllWindows()
if__name__=="__main__":
resnet_demo()
審核編輯 :李倩
聲明:本文內容及配圖由入駐作者撰寫或者入駐合作網站授權轉載。文章觀點僅代表作者本人,不代表電子發燒友網立場。文章及其配圖僅供工程師學習之用,如有內容侵權或者其他違規問題,請聯系本站處理。
舉報投訴
-
模型
+關注
關注
1文章
3176瀏覽量
48721 -
腳本
+關注
關注
1文章
387瀏覽量
14833
原文標題:OpenVINO2022 Dev Tools安裝與使用
文章出處:【微信號:CVSCHOOL,微信公眾號:OpenCV學堂】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
相關推薦
嵌入式學習-搭建自己的ubuntu系統之VMware Tools安裝
VMware Tools安裝上一節完成了ubuntu系統的安裝,接下來安裝VMware Tools,如果不
發表于 08-13 08:57
OpenVINO2024 C++推理使用技巧
很多人都使用OpenVINO新版的C++ 或者Python的SDK,都覺得非常好用,OpenVINO2022之后的版本C++ SDK做了大量的優化與整理,已經是非常貼近開發的使用習慣與推理方式。與OpenCV的Mat對象對接方式更是幾乎無縫對接,非常的方便好用。
使用esp-idf-tools2.12不能搭建esp32s3開發環境是什么原因?
下載了最新的esp-idf-tools2.12安裝工具,該工具安裝過程中,esp-idf版本只能選擇V4.3.1和V4.2.2,沒有V4.4版本選項(安裝指導文檔建議選擇該版本),如附
發表于 06-27 08:20
使用ESP-IDF工具安裝器esp-idf-tools-setup-2.2.exe進行安裝卡住了是怎么回事?
我在使用ESP-IDF 工具安裝器esp-idf-tools-setup-2.2.exe進行安裝的時候,執行到下圖的位置,就一直停留在這里,沒有繼續,只有Cancel。
我試著復制命令到cmd下執行,出現如下提示:
麻煩大家幫
發表于 06-25 07:45
IDF 3.3.1 UBUNTU下安裝報錯的原因?
操作系統為Ubuntu 18.04.4 LTS,64位
按照安裝指導成功安裝了準備工作,
sudo apt-get install git wget flex bison gperf python
發表于 06-25 07:32
esp-idf tools安裝時總是發生異常怎么處理?
請問,在ESP-IDF tools 安裝中,總是在提示發生exception, log框中顯示fatal python error: 見下圖:
請問是什么問題,怎么解決,多謝!
發表于 06-21 08:27
windows下idf-tools安裝進行到 “Switching branch”時失敗了的原因?
windows下idf-tools安裝進行到 “Switching branch”時失敗
安裝時選擇的版本為4.2
4.3版本可以正常安裝
發表于 06-21 07:19
linux系統已安裝esp-idf,無法安裝vscode插件要怎么處理?
deepin/linux系統,使用第三種方式安裝
無法add openocd dev rules
EISDIR: illegal operation on a directory
發表于 06-20 07:39
為OpenVINO添加對Paddle 2.5的支持
我是飛槳黑客馬拉松第五期 OpenVINO 賽題獲獎者——為 OpenVINO 添加了對 Paddle 2.5 的支持。在此記錄下來貢獻的過程,希望有更多的同學可以參與到 OpenVINO 的社區
基于OpenVINO?和AIxBoard的智能安檢盒子設計
公共安全問題日益凸顯,傳統安檢手段面臨挑戰。為提高安檢效率和保障公共安全,基于 OpenVINO? 和 AIxBoard 的智能安檢盒子應運而生。
全志V853 NPU開發之工具安裝
;./Vivante_IDE-5.7.0_CL470666-Linux-x86_64-04-24-2022-18.55.31-plus-W-p6.4.x_dev_6.4.10_22Q1_CL473325A-Install
單擊Next
閱讀許可協議并允許。
選擇安裝路
發表于 01-08 09:50
如何快速下載OpenVINO Notebooks中的AI大模型
OpenVINO Notebooks是Jupyter Notebook形式的OpenVINO范例程序大集合,方便開發者快速學習并掌握OpenVINO推理程序,并通過Copy&Paste方式將范例中的關鍵程序應用到自己的AI軟件中
評論