First argument as config.env is mandatory
This commit is contained in:
1 parent
d66af69c42
commit
ccb1c89b12
11 files changed
+118
-3
No files matched your search
+25
-2
@@ -5,6 +5,7 @@
|
||||
#include "rknn_yolo.hpp"
|
||||
#include "drawing.hpp"
|
||||
#include "nms.hpp"
|
||||
#include "version.hpp"
|
||||
|
||||
#include <opencv2/opencv.hpp>
|
||||
#include <Eigen/Dense>
|
||||
@@ -98,10 +99,30 @@ int main(int argc, char* argv[]) {
|
||||
signal(SIGINT, signal_handler);
|
||||
signal(SIGTERM, signal_handler);
|
||||
|
||||
const char* env_path = argc > 1 ? argv[1] : nullptr;
|
||||
g_config.load_from_env(env_path);
|
||||
if (argc < 2) {
|
||||
std::cout << "ByteTrack Counter (C++)" << std::endl;
|
||||
std::cout << "Version: " << version::GIT_COMMIT
|
||||
<< " | branch: " << version::GIT_BRANCH
|
||||
<< " | built: " << version::BUILD_DATE << " " << version::BUILD_TIME
|
||||
<< " | rknn: " << (version::BUILD_RKNN ? "ON" : "OFF");
|
||||
if (!version::GIT_TAG.empty())
|
||||
std::cout << " | tag: " << version::GIT_TAG;
|
||||
std::cout << std::endl;
|
||||
std::cout << "Usage: bytetrack-counter <config.env>" << std::endl;
|
||||
std::cout << " config.env path to environment configuration file (required)" << std::endl;
|
||||
return 1;
|
||||
}
|
||||
|
||||
g_config.load_from_env(argv[1]);
|
||||
|
||||
std::cout << "Starting ByteTrack Counter (C++)" << std::endl;
|
||||
std::cout << "Version: " << version::GIT_COMMIT
|
||||
<< " | branch: " << version::GIT_BRANCH
|
||||
<< " | built: " << version::BUILD_DATE << " " << version::BUILD_TIME
|
||||
<< " | rknn: " << (version::BUILD_RKNN ? "ON" : "OFF");
|
||||
if (!version::GIT_TAG.empty())
|
||||
std::cout << " | tag: " << version::GIT_TAG;
|
||||
std::cout << std::endl;
|
||||
std::cout << "Camera: " << g_config.camera_name << std::endl;
|
||||
std::cout << "Source: " << g_config.source << std::endl;
|
||||
std::cout << "Model: " << g_config.model_path << " | imgsz=" << g_config.imgsz
|
||||
@@ -244,6 +265,8 @@ int main(int argc, char* argv[]) {
|
||||
int na = static_cast<int>(ayam_boxes.size());
|
||||
int nt = static_cast<int>(talenan_boxes.size());
|
||||
|
||||
if (na > 0) store.ping_activity();
|
||||
|
||||
Eigen::MatrixXf ayam_mat(na, 4), talenan_mat(nt, 4);
|
||||
Eigen::VectorXf ayam_scores_vec(na), talenan_scores_vec(nt);
|
||||
|
||||
|
||||
Reference in new issue
Block a user