1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798 |
- #ifndef MODULES_INFERENCE2_INCLUDE_VIDEO_PREPROC_HPP_
- #define MODULES_INFERENCE2_INCLUDE_VIDEO_PREPROC_HPP_
- #include <memory>
- #include <string>
- #include <utility>
- #include <vector>
- #include "cnis/contrib/video_helper.h"
- #include "cnis/infer_server.h"
- #include "cnis/processor.h"
- #include "cnstream_frame.hpp"
- #include "cnstream_frame_va.hpp"
- #include "reflex_object.h"
- namespace cnstream {
- class VideoPreproc : virtual public ReflexObjectEx<VideoPreproc> {
- public:
-
- virtual ~VideoPreproc() {}
-
- static VideoPreproc* Create(const std::string& proc_name);
-
- void SetModelInputPixelFormat(infer_server::video::PixelFmt fmt) {
- model_input_pixel_format_ = fmt;
- }
-
- virtual bool Execute(infer_server::ModelIO* model_input, const infer_server::InferData& input_data,
- const infer_server::ModelInfo& model_info) = 0;
- protected:
- infer_server::video::PixelFmt model_input_pixel_format_ = infer_server::video::PixelFmt::RGB24;
- };
- }
- #endif
|