fix bug when used as lib
This commit is contained in:
parent
cc715bef70
commit
9612f9fe83
|
@ -9,7 +9,7 @@
|
|||
#endif
|
||||
|
||||
#ifdef WITH_INTEL
|
||||
#include<openvino/openvino.hpp>
|
||||
#include <openvino/openvino.hpp>
|
||||
#include "common_det_intel_impl.h"
|
||||
#endif
|
||||
|
||||
|
|
|
@ -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);
|
||||
|
|
|
@ -25,7 +25,6 @@ public:
|
|||
double line_location_a2;
|
||||
|
||||
bool is_load_parameter;
|
||||
|
||||
std::string line_color;
|
||||
|
||||
protected:
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
};
|
||||
|
||||
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue