123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263 |
- #include "UVCGrabber/UVCDeviceManager.h"
- // #include "RTSPGrabber/InfraredManager.hpp"
- #include "MppDecoder/MppManager.h"
- #include "AIManager/RKNNManager.h"
- #include "DataManager/DataManager.h"
- #include "DataManager/DataPackage.h"
- #include "ImageTest/ImageTest.h"
- #include <stdexcept>
- #include <iostream>
- #include <json.hpp>
- #define DEVICE_NUM 3
- int main(int argc, char **argv)
- {
- // InfraredManager Infrar;
- // Infrar.init("rtsp://admin:admin123@192.168.211.108");
- // Infrar.openStream();
- std::system("echo 64 > /sys/module/usbcore/parameters/usbfs_memory_mb");
- std::system("rm ./TimeLog.txt");
- // use json to read config file
- nlohmann::json config;
- try {
- std::ifstream config_file("config.json");
- config_file >> config;
- config_file.close();
- } catch (std::exception &e) {
- std::cerr << "Error reading config file: " << e.what() << std::endl;
- return 1;
- }
- UsbTest::TimeRecorder::getInstance().init();
- UsbTest::GlobalResolutionTimer::getInstance().start();
- UsbTest::SimulateTrigger::getInstance().startTrigger();
- // Initialize the UVC manager
- UVCManager::getInstance().init();
- MppManager::getInstance().addMppDecode();
- RKNNManager::getInstance().setGrpcServerAddress(config["grpc_server_address"]);
- RKNNManager::getInstance().addRknnTask("../ppyoloe.rknn");
- // Start all streaming
- UVCManager::getInstance().startAllStreaming();
- // Wait for the user to press a key
- std::cin.get();
- RKNNManager::getInstance().StopThread();
- // Stop all streaming
- UVCManager::getInstance().stopAllStreaming();
- return 0;
- }
|