first commit

This commit is contained in:
陈赣
2026-06-03 12:43:14 +08:00
commit ba76cfae28
608 changed files with 120791 additions and 0 deletions

93
objects/vp_frame_meta.h Executable file
View File

@@ -0,0 +1,93 @@
#pragma once
#include <vector>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/videoio.hpp>
#include "vp_meta.h"
#include "vp_frame_target.h"
#include "vp_frame_pose_target.h"
#include "vp_frame_face_target.h"
#include "vp_frame_text_target.h"
#include "ba/vp_ba_result.h"
/*
* ##########################################
* how does frame meta work?
* ##########################################
* frame meta, holding all data(targets/elements/...) of current frame in the video scene. frame meta are independent and don't know about each other, neither its previous frames nor next frames.
* the data in frame meta is just telling us what **current frame** is so we can not get something like 'state-switch' from a single frame meta.
* if you need know when the 'state-switch' happen, for example, you want to notify to cloud via restful api if state changed(ignore if it's keeping),
* you need cache previous frame meta(maybe partial data) in your custom node first and then compare with each other to figure out if it has changed.
*
* frame meta works like our eyes, by taking a glance at the frame in video we can see what the picture is and how many targets are there.
* but if you want to know something like state-switch, for example, a person was walking and then stop or it stop for a while and then start to walk, you have to see(cache) more frames.
*
* see more implementation of 'vp_track_node' and 'vp_message_broker_node' which saved history frame meta data and then work based on them.
* 1. vp_track_node : save previous locations of targets and then do tracking based on them, we need see more frames to track targets in video.
* 2. vp_message_broker_node : save previous ba_flags and then do notifying based on them, we need see more frames to check if state-switch has happened.
* ##########################################
*/
namespace vp_objects {
// frame meta, which contains frame-related data. it is kind of important meta in pipeline.
class vp_frame_meta: public vp_meta {
private:
/* data */
public:
vp_frame_meta(cv::Mat frame, int frame_index = -1, int channel_index = -1, int original_width = 0, int original_height = 0, int fps = 0);
~vp_frame_meta();
// define copy constructor since we need deep copy operation.
vp_frame_meta(const vp_frame_meta& meta);
// frame the meta belongs to, filled by src nodes.
int frame_index;
// fps for current video.
int fps;
// orignal frame width, fiiled by src nodes.
int original_width;
// original frame height, filled by src nodes.
int original_height;
// image data the meta holds, filled by src nodes.
// deep copy needed here for this member.
cv::Mat frame;
// osd image data the meta holds, filled by osd node if exists.
// deep copy needed here for this member.
cv::Mat osd_frame;
// mask for the WHOLE frame, filled by Semantic Segmentation nodes if exists.
// deep copy needed here for this member.
cv::Mat mask;
// text description for frame (output from LLM)
std::string description;
// targets created/appended by primary infer nodes, and then updated by secondary infer nodes if exist.
// it is shared_ptr<...> type just to keep same as elements.
// deep copy needed here for this member.
std::vector<std::shared_ptr<vp_objects::vp_frame_target>> targets;
// pose targets created/appened by primary infer nodes.
std::vector<std::shared_ptr<vp_objects::vp_frame_pose_target>> pose_targets;
// face targets created/appened by primary infer nodes.
std::vector<std::shared_ptr<vp_objects::vp_frame_face_target>> face_targets;
// text targets created/appened by primary infer nodes.
std::vector<std::shared_ptr<vp_objects::vp_frame_text_target>> text_targets;
// ba results created/appened by ba nodes.
std::vector<std::shared_ptr<vp_objects::vp_ba_result>> ba_results;
// get target ptrs by target ids in current frame, ONLY supports vp_frame_target
std::vector<std::shared_ptr<vp_frame_target>> get_targets_by_ids(const std::vector<int>& ids);
// copy myself
virtual std::shared_ptr<vp_meta> clone() override;
};
}