VERI test Edition

This commit is contained in:
lxm
2023-09-12 10:33:16 +08:00
parent 0c489ad9cb
commit 6d7d666e2c
9 changed files with 578 additions and 0 deletions

36
include/sv_veri_det.h Normal file
View File

@@ -0,0 +1,36 @@
#ifndef __SV_VERI_DET__
#define __SV_VERI_DET__
#include "sv_core.h"
#include <opencv2/opencv.hpp>
#include <opencv2/aruco.hpp>
#include <opencv2/tracking.hpp>
#include <string>
#include <chrono>
namespace sv {
class VeriDetectorCUDAImpl;
class VeriDetector : public LandingMarkerDetectorBase
{
public:
VeriDetector();
~VeriDetector();
void detect(cv::Mat img1_, cv::Mat img2_, TargetsInFrame &tgts_);
protected:
bool setupImpl();
void roiCNN(
std::vector<cv::Mat>& input_rois_,
std::vector<int>& output_labels_
);
VeriDetectorCUDAImpl* _cuda_impl;
};
}
#endif

View File

@@ -7,6 +7,7 @@
#include "sv_sot.h"
#include "sv_mot.h"
#include "sv_color_line.h"
#include "sv_veri_det.h"
#include "sv_video_input.h"
#include "sv_video_output.h"