fix bug when used as lib

This commit is contained in:
jario 2024-01-02 14:02:34 +08:00
parent cc715bef70
commit 9612f9fe83
6 changed files with 54 additions and 81 deletions

View File

@ -9,7 +9,7 @@
#endif
#ifdef WITH_INTEL
#include<openvino/openvino.hpp>
#include <openvino/openvino.hpp>
#include "common_det_intel_impl.h"
#endif

View File

@ -1002,11 +1002,11 @@ void CommonObjectDetectorBase::detect(cv::Mat img_, TargetsInFrame& tgts_, Box*
if (this->_with_segmentation)
{
cv::Mat mask_j = boxes_seg[j].clone();
#ifdef WITH_INTEL
#ifdef WITH_INTEL
tgt.setMask(mask_j);
#endif
#endif
#ifdef WITH_CUDA
#ifdef WITH_CUDA
int maskh = mask_j.rows, maskw = mask_j.cols;
assert(maskh == maskw);
@ -1036,7 +1036,7 @@ void CommonObjectDetectorBase::detect(cv::Mat img_, TargetsInFrame& tgts_, Box*
{
tgt.setMask(mask_j);
}
#endif
#endif
}
tgts_.targets.push_back(tgt);

View File

@ -25,7 +25,6 @@ public:
double line_location_a2;
bool is_load_parameter;
std::string line_color;
protected:

View File

@ -10,21 +10,18 @@
namespace sv
{
#ifdef WITH_CUDA
class CommonObjectDetectorCUDAImpl;
#endif
#ifdef WITH_INTEL
class CommonObjectDetectorIntelImpl;
#endif
class CommonObjectDetectorCUDAImpl;
class CommonObjectDetectorIntelImpl;
class CommonObjectDetector : public CommonObjectDetectorBase
{
public:
class CommonObjectDetector : public CommonObjectDetectorBase
{
public:
CommonObjectDetector(bool input_4k = false);
~CommonObjectDetector();
protected:
protected:
bool setupImpl();
void detectImpl(
cv::Mat img_,
@ -36,16 +33,11 @@ namespace sv
std::vector<float> &boxes_score_,
std::vector<cv::Mat> &boxes_seg_);
#ifdef WITH_CUDA
CommonObjectDetectorCUDAImpl *_cuda_impl;
#endif
#ifdef WITH_INTEL
CommonObjectDetectorIntelImpl *_intel_impl;
#endif
bool _input_4k;
};
};
}
#endif

View File

@ -10,13 +10,9 @@
namespace sv {
#ifdef WITH_CUDA
class LandingMarkerDetectorCUDAImpl;
#endif
#ifdef WITH_INTEL
class LandingMarkerDetectorIntelImpl;
#endif
class LandingMarkerDetectorCUDAImpl;
class LandingMarkerDetectorIntelImpl;
class LandingMarkerDetector : public LandingMarkerDetectorBase
{
@ -30,13 +26,8 @@ protected:
std::vector<int>& output_labels_
);
#ifdef WITH_CUDA
LandingMarkerDetectorCUDAImpl *_cuda_impl;
#endif
#ifdef WITH_INTEL
LandingMarkerDetectorIntelImpl *_intel_impl;
#endif
};

View File

@ -10,23 +10,19 @@
namespace sv
{
#ifdef WITH_CUDA
class VeriDetectorCUDAImpl;
#endif
#ifdef WITH_INTEL
class VeriDetectorIntelImpl;
#endif
class VeriDetectorCUDAImpl;
class VeriDetectorIntelImpl;
class VeriDetector : public LandingMarkerDetectorBase
{
public:
class VeriDetector : public LandingMarkerDetectorBase
{
public:
VeriDetector();
~VeriDetector();
void detect(cv::Mat img_, const cv::Rect &bounding_box_, sv::Target &tgt);
protected:
protected:
void _load();
bool setupImpl();
void roiCNN(
@ -40,14 +36,9 @@ namespace sv
std::vector<float> targetSz = {0, 0}; // H and W of bounding box
std::vector<float> targetPos = {0, 0}; // center point of bounding box (x, y)
#ifdef WITH_CUDA
VeriDetectorCUDAImpl *_cuda_impl;
#endif
#ifdef WITH_INTEL
VeriDetectorIntelImpl *_intel_impl;
#endif
};
};
}
#endif