| | |
| | | #include "../nodes/vp_file_src_node.h"
|
| | | #include "../nodes/vp_rtsp_src_node.h"
|
| | | #include "../nodes/vp_rtsp_ffmpeg_src_node.h"
|
| | |
|
| | | #include "../nodes/infers/vp_yolo_detector_node.h"
|
| | | #include "../nodes/track/vp_sort_track_node.h"
|
| | |
| | | rtsp_path = args[0];
|
| | | resize_ratio = std::stof(args[1]);
|
| | | kafka_server_point = args[2];
|
| | | |
| | | } else {
|
| | | std::cout << "Usage: " << argv[0] << " <rtsp_path> <resize_ratio> <kafka_server_point>\n";
|
| | | std::cout << "Example: " << argv[0] << " rtsp://192.168.1.24:8554/demo 0.4 192.168.130.228:9092\n";
|
| | |
| | |
|
| | | // create nodes
|
| | | //auto file_src_0 = std::make_shared<vp_nodes::vp_file_src_node>("file_src_0", 0, video_path, resize_ratio);
|
| | | auto rtsp_src_1 = std::make_shared<vp_nodes::vp_rtsp_ffmpeg_src_node>("rtsp_src_1", 0, rtsp_path, resize_ratio,0,true); |
| | | auto rtsp_src_1 = std::make_shared<vp_nodes::vp_rtsp_src_node>("rtsp_src_1", 0, rtsp_path, resize_ratio, "avdec_h264"); |
| | | auto yolo_detector = std::make_shared<vp_nodes::vp_yolo_detector_node>("yolo_detector", "./vp_data/models/det_cls/yolov3-tiny-2022-0721_best.weights", "./vp_data/models/det_cls/yolov3-tiny-2022-0721.cfg", "./vp_data/models/det_cls/yolov3_tiny_5classes.txt");
|
| | | auto tracker = std::make_shared<vp_nodes::vp_sort_track_node>("sort_tracker");
|
| | |
|