floraachy
/
SpireCV
forked from floratest1/SpireCV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
algorithm/common_det/sv_common_det.cpp
View File

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

8
algorithm/sv_algorithm_base.cpp
View File

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

1
include/sv_color_line.h
View File

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

24
include/sv_common_det.h
View File

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

13
include/sv_landing_det.h
View File

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

23
include/sv_veri_det.h
View File

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