老百姓的日常餐桌上總少不了魚,大快朵頤背后,您是否想到過,這些海產品通常都經過水產養殖企業的科學養殖處理,且肉感鮮美,安全可靠。
然而,水產養殖企業的大多數工作仍舊大量依靠人力,工人需要在潮濕環境下進行高強度的漁業作業,造成了鮮有年輕人愿意從事這類繁重、惡劣的工作。因此實現水產養殖的自動化流程,對水產養殖企業來說是極大的需求。
水產養殖企業的一個主要賣品就是魚,包含各種魚類的魚苗和成魚。魚苗通常會被進行飼養,直到養成成魚后再賣出。成魚則是會被放進池塘中進行觀察飼養。由于魚苗的個體小,在一個水盆里很難判斷魚苗的數量,因此需要通過視覺目標檢測的方法來對魚苗計數。在國內、國外都有成熟的魚苗計數器來對魚苗的數量進行計數,有較高的準確率,但是卻很少有成熟的、針對成魚的計數產品。
來自華南理工大學自動化科學與工程學院的研究生林銳描述說:“一方面,成魚的活動性要比魚苗強得多,使得成魚在不同時刻可能有不同的姿態,給視覺檢測造成困擾;另一方面,成魚之間相互重疊遮擋的情況更為明顯,較難從二維圖像中判斷魚遮擋重疊時的數量情況。”
用實例分割模型對成魚計數
作為校企合作項目,林銳和他的小伙伴們一起,與某水產養殖企業合作,目標是幫助企業在放魚作業時能了解魚的數量和大致重量。
本項目的應用場景是:水產養殖企業將采購的魚裝在車廂中,并載回到養殖場,將車廂和放魚槽連接,放魚槽另一端連接池塘,放魚槽上安裝了本項目的魚計數和估重系統。打開車廂的閘門,將魚放進放魚槽并流向池塘,該系統則開始對魚進行計數和估重,并將結果上傳到后臺服務器中。
圖1放魚槽示意圖
圖 1 中的 ① 即為NVIDIAJetson Xavier NX設備,② 為攝像頭,① 和 ② 組成了本項目的硬件設備。③ 為該項目的實驗對象。④ 為該項目所需要的放魚槽設備。
林銳和他的隊員們決定,使用深度學習實例分割算法作為魚的目標檢測模型。“實例分割是語義分割和目標檢測的集成,它具有目標檢測的功能,同時能對目標進行像素級的分割。”林銳介紹說。
利用圖像中的一些語義信息,來幫助模型對相互遮擋重疊的魚進行識別,從而使算法能更好地對成魚進行計數。利用實例分割模型得到的目標掩膜,可以建立目標像素面積和重量的一個線性模型,通過這個模型得到魚的估計重量,幫助企業能較好地了解這一批次魚的數量和重量。
在算力和成本中達到最佳平衡
同學們利用現場收集的上千張魚的圖片作為訓練數據集,部分圖片如圖 2 所示。
圖 2訓練集圖片
利用收集的魚類數據集訓練算法,使用了 PyTorch1.4 框架,CUDA10.0以及NVIDIA GPU進行訓練,并且算法的 Backbone 層修改為 ResNet18。在訓練完算法之后,先將模型轉化為 ONNX 格式,將模型權重導出為 wts 格 式,并通過TensorRT框架實現 YOLACT 算法的加速引擎。部署到 Jetson Xavier NX 上使用的是 C++ 語言,然后使用了 Qt 開源框架搭建了一個人機操作界面,并且移植到 Jetson Xavier NX 平臺上。
“由于本項目用到實例分割算法,需要算力較高的設備。”林銳說,權衡了成本后,最終選擇 NVIDIA Jetson Xavier NX。它擁有 384 個 CUDA 核心以及 48 個Tensor核 心,Volta架構,在 INT8 數據格式下可以達到 21TOPS 的算力,在 FP16 數據格式下可以達到 10TOPS 的算力。
將整個 Qt 程序移植到 NVIDIA Jetson Xavier NX 設備上,最后的呈現效果如圖 3 所 示。
圖 3Jetson Xavier NX 界面顯示
“雖然本項目主要研究成魚的計數和估重,只是水產養殖業中的一個簡單環節, 但是本項目的技術可以推廣到水產養殖業的其他工作任務,”林銳說,“國內的水產養殖企業大多是勞動密集型企業,大多數工作需要人力來完成。因此,通過引進一些智能科技,幫助水產養殖企業打造自動化流水線,能提高水產養殖企業的經濟收益和工作效率。這將有利于原型機產品的推廣和研發,并打開一個新的市場。”
給 Jetson 開發者們的建議
作為資深開發者,已經使用了好幾代 NVIDIA Jetson 產品的林銳說:“Jetson 系列有多個不同型號的設備,適用于不同的應用場景和需求。因此,在選擇 Jetson 設備時,需要根據自己的需求選擇合適的設備。”
“作為 NVIDIA Jetson 用戶,需要深入了解設備的功能和性能,學習優化代碼以實現更好的性能。與此同時,Jetson 擁有一個活躍的社區,包括開發者、用戶和愛好者。加入社區可以獲得幫助、分享經驗和學習新技能。Jetson 社區還有許多活動和比賽,可以幫助用戶提高技能,認識到更多的應用場景。”
審核編輯 :李倩
-
NVIDIA
+關注
關注
14文章
5052瀏覽量
103364 -
自動化
+關注
關注
29文章
5603瀏覽量
79470 -
數據集
+關注
關注
4文章
1208瀏覽量
24749
原文標題:Jetson 百萬開發者故事 | NVIDIA Jetson 助力水產養殖企業打造自動化流水線
文章出處:【微信號:NVIDIA_China,微信公眾號:NVIDIA英偉達】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
相關推薦
評論