45 lines
2.4 KiB
C++
Executable File
45 lines
2.4 KiB
C++
Executable File
|
|
#include "vp_primary_infer_node.h"
|
|
|
|
namespace vp_nodes {
|
|
|
|
vp_primary_infer_node::vp_primary_infer_node(std::string node_name,
|
|
std::string model_path,
|
|
std::string model_config_path,
|
|
std::string labels_path,
|
|
int input_width,
|
|
int input_height,
|
|
int batch_size,
|
|
int class_id_offset,
|
|
float scale,
|
|
cv::Scalar mean,
|
|
cv::Scalar std,
|
|
bool swap_rb,
|
|
bool swap_chn):
|
|
vp_infer_node(node_name,
|
|
vp_infer_type::PRIMARY,
|
|
model_path,
|
|
model_config_path,
|
|
labels_path,
|
|
input_width,
|
|
input_height,
|
|
batch_size,
|
|
scale,
|
|
mean,
|
|
std,
|
|
swap_rb,
|
|
swap_chn),
|
|
class_id_offset(class_id_offset) {
|
|
}
|
|
|
|
vp_primary_infer_node::~vp_primary_infer_node() {
|
|
|
|
}
|
|
|
|
void vp_primary_infer_node::prepare(const std::vector<std::shared_ptr<vp_objects::vp_frame_meta>>& frame_meta_with_batch, std::vector<cv::Mat>& mats_to_infer) {
|
|
// fetch the whole frame, can batch by batch
|
|
for (auto& i: frame_meta_with_batch) {
|
|
mats_to_infer.push_back(i->frame);
|
|
}
|
|
}
|
|
} |