It is for my undergrad thesis in Tsinghua University.
There are four modules in the project:
A Libtorch implementation of the YOLO v3 object detection algorithm, written with modern C++.
The code is based on the walktree.
The config file in .\models can be found at Darknet.
I also merged SORT to do tracking.
Currently on a GTX 1060 6G it consumes about 1G RAM and have 37 FPS.
The video I test is TownCentreXVID.avi.
With wxWidgets, I developed the GUI module for visualization of results.
Previously I used Dear ImGui. However, I do not think it suits my purpose.
LibTorch can be easily integrated with CMake, but there are a lot of strange things...
On Ubuntu 16.04, I use
apt installto install the others. Everything is fine. On Windows 10 + Visual Studio 2017, I use the latest stable version of the others from their official websites.
Here are some intermediate output from detection and tracking module:
Here is the snapshot of processing module:
Here is the snapshot of GUI module: