libtorch yolov3

libtorch-yolov3

YOLO v3 物件偵測演算法的 Libtorch 實現，以純 C++ 編寫。它可以快速、輕鬆地整合到您的製作中，並且支援 CPU 和 GPU。欣賞吧~

這個專案受到pytorch版本的啟發，我用C++重寫了它。

要求

1. LibTorch v1.0.0
2. 庫達
3. OpenCV（僅在範例中使用）

編譯

1. cmake3
2. 海灣合作委員會 5.4 +

 mkdir build && cd build
cmake3 -DCMAKE_PREFIX_PATH="your libtorch path" ..

# if there are multi versions of gcc, then tell cmake which one your want to use, e.g.:
cmake3 -DCMAKE_PREFIX_PATH="your libtorch path" -DCMAKE_C_COMPILER=/usr/local/bin/gcc -DCMAKE_CXX_COMPILER=/usr/local/bin/g++ ..

運行檢測器

您需要做的第一件事是取得 v3 的權重檔案：

 cd models
wget https://pjreddie.com/media/files/yolov3.weights 

在單張圖像上：

 ./yolo-app ../imgs/person.jpg

據我測試，在 GPU (1080 ti) 上需要 25 毫秒。請多次執行推理作業，計算平均成本。