在DeepStream上使用自己的Pytorch模型

Posted By Lin Felix on 9 月 23, 2021 in DeepStream, Edge AI開發地圖, Edge AI開發環境, Jetson Nano, NVIDIA, PyTorch, 教學文 | 0 comments

DeepStream是NVIDIA專為處理多個串流影像，並進行智慧辨識而整合出的強大工具。開發語言除了原先的C++，從DeepStream SDK 5.1也支援基於原先安裝，再掛上Python套件的方式，讓較熟悉Python程式語言的使用者也能使用DeepStream。

本文主要將其應用在Jetson Nano上，並於DeepStream導入自己的模型執行辨識。

在Jetson Nano上面安裝DeepStream

筆者使用的硬體為Jetson Nano 2GB/4GB，參照官方提供的步驟與對應的版本，幾乎可以說是無痛安裝。對比同樣採用乾淨映像檔，使用源碼或是Docker安裝的JetBot與Jetson Inference要快上許多。