前言
之前每次進(jìn)行機(jī)器學(xué)習(xí)和模型訓(xùn)練的時(shí)候發(fā)現(xiàn)想要訓(xùn)練不同模型的時(shí)候需要使用不同的框架,有時(shí)候費(fèi)了九牛二虎之力終于寫(xiě)下了幾百行代碼之后,才發(fā)現(xiàn)環(huán)境調(diào)試不通,運(yùn)行效率也差強(qiáng)人意,于是自己寫(xiě)了一個(gè)基于LabVIEW的機(jī)器視覺(jué)工具包,讓編程變得更簡(jiǎn)單便捷的同時(shí),還能夠使用多種框架和硬件加速。
一、工具包內(nèi)容
此人工智能視覺(jué)工具包主要優(yōu)勢(shì)如下:
- 圖形化編程,無(wú)需掌握文本編程基礎(chǔ)即可完成機(jī)器視覺(jué)項(xiàng)目。
- 多種攝像頭數(shù)據(jù)采集和矩陣計(jì)算。
- 數(shù)百種圖像算子的調(diào)用。
- 提供tensorflow、pytorch、caffe、darknet、onnx、paddle等多種框架深度學(xué)習(xí)模型的調(diào)用并實(shí)現(xiàn)推理。
- 支持Nvidia GPU、Intel、TPU、NPU多種加速。
- 提供近百個(gè)應(yīng)用程序范例,包括物體分類(lèi)、物體檢測(cè)、物體測(cè)量、圖像分割、 人臉識(shí)別、自然場(chǎng)景下OCR等多種實(shí)用場(chǎng)景。**
工具包中的函數(shù)選版如下:
例如,一個(gè)攝像頭采集并進(jìn)行yolov5目標(biāo)檢測(cè)的范例程序,只需在LabVIEW中編寫(xiě)簡(jiǎn)單的圖形化程序,即可實(shí)現(xiàn)。在大量簡(jiǎn)化編程難度的同時(shí),也保持了c++的高效運(yùn)行特性。
?
二、工具包下載鏈接
https://pan.baidu.com/s/1nyclNG8uMLnYBmcGKVDtWQ?pwd=yiku
三、實(shí)現(xiàn)物體識(shí)別
無(wú)論使用何種框架訓(xùn)練物體檢測(cè)模型,都可以無(wú)縫集成到LabVIEW中,并使用智能視覺(jué)工具包提供的CUDA接口實(shí)現(xiàn)加速推理,模型包括但不限于:
- yolov3/yolov4/yolov5//yolox/pp-yolo
- SSD,F(xiàn)astest-RCNN(物體檢測(cè))
- mobileNet、VGGnet、Resnet、Densenet、Efficientnet等(物體分類(lèi))
通過(guò)算法優(yōu)化,在LabVIEW中運(yùn)行模型的速度明顯好于python,這對(duì)于對(duì)性能要求較高的工業(yè)現(xiàn)場(chǎng)來(lái)說(shuō)非常友好實(shí)用。比如說(shuō):工地安全帽檢測(cè)、物體表面缺陷檢測(cè)等,如下圖使用yolov4進(jìn)行物體識(shí)別,在GPU模式下,無(wú)論是運(yùn)行速度和識(shí)別率都可以達(dá)到工業(yè)級(jí)別。。
實(shí)測(cè)過(guò)程中我們發(fā)現(xiàn)同一系統(tǒng)環(huán)境下,使用labview工具包的識(shí)別效率遠(yuǎn)高于python識(shí)別效率。
四、實(shí)現(xiàn)圖像分割
圖像分割是當(dāng)今計(jì)算機(jī)視覺(jué)領(lǐng)域的關(guān)鍵問(wèn)題之一。從宏觀上看,圖像分割是一項(xiàng)高層次的任務(wù),為實(shí)現(xiàn)場(chǎng)景的完整理解鋪平了道路。場(chǎng)景理解作為一個(gè)核心的計(jì)算機(jī)視覺(jué)問(wèn)題,其重要性在于越來(lái)越多的應(yīng)用程序通過(guò)從圖像中推斷知識(shí)來(lái)提供營(yíng)養(yǎng)。隨著深度學(xué)習(xí)軟硬件的加速發(fā)展,一些前沿的應(yīng)用包括自動(dòng)駕駛汽車(chē)、人機(jī)交互、醫(yī)療影像等,都開(kāi)始研究并使用圖像分割技術(shù)。
本次集成的智能工具包提供了多種圖像分割的調(diào)用模塊,并實(shí)現(xiàn)了GPU模式下的加速運(yùn)行。如:
語(yǔ)義分割:Segnet、deeplabv1~deeplabv3、u-net等;實(shí)例分割:Mask-RCNN、PANet等
五、自然場(chǎng)景下的文字識(shí)別
人工智能提供了文本檢測(cè)定位(DB_TD500_resnet50、EAST)、文本識(shí)別的模塊(CRNN),用戶(hù)可以使用該模塊實(shí)現(xiàn)自然場(chǎng)景下的中英文文字識(shí)別
應(yīng)用:身份證識(shí)別、表單識(shí)別、包裝盒標(biāo)簽檢測(cè)等
總結(jié)
可以通過(guò)鏈接進(jìn)行工具包的下載,如有問(wèn)題可添加技術(shù)交流群進(jìn)行進(jìn)一步的探討。
審核編輯 黃宇
-
LabVIEW
+關(guān)注
關(guān)注
1976文章
3657瀏覽量
324816 -
AI
+關(guān)注
關(guān)注
87文章
31399瀏覽量
269787 -
人工智能
+關(guān)注
關(guān)注
1793文章
47590瀏覽量
239472 -
深度學(xué)習(xí)
+關(guān)注
關(guān)注
73文章
5511瀏覽量
121382
發(fā)布評(píng)論請(qǐng)先 登錄
相關(guān)推薦
評(píng)論