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 毫秒。请多次运行推理作业，并计算平均成本。