Add CenterTrack pre, post, visualization and demo.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Davide Sapienza
2020-12-07 18:45:29 +01:00
parent 9f10c3f6e2
commit 48ecebe6dd
3 changed files with 1043 additions and 0 deletions
+5
View File
@@ -5,6 +5,7 @@
#include <mutex>
#include "CenternetDetection3D.h"
#include "CenternetDetection3DTrack.h"
bool gRun;
bool SAVE_RESULT = false;
@@ -34,6 +35,7 @@ int main(int argc, char *argv[]) {
n_classes = atoi(argv[4]);
tk::dnn::CenternetDetection3D cnet;
tk::dnn::CenternetDetection3DTrack ctrack;
tk::dnn::DetectionNN3D *detNN;
@@ -42,6 +44,9 @@ int main(int argc, char *argv[]) {
case 'c':
detNN = &cnet;
break;
case 't':
detNN = &ctrack;
break;
default:
FatalError("Network type not allowed (3rd parameter)\n");
}