Net color depth is exposed from Detector class

pull/981/head
IlyaOvodov 7 years ago
parent ec766fc3f1
commit 5291935b70
  1. 4
      src/yolo_v2_class.cpp
  2. 1
      src/yolo_v2_class.hpp

@ -119,6 +119,10 @@ YOLODLL_API int Detector::get_net_height() const {
detector_gpu_t &detector_gpu = *static_cast<detector_gpu_t *>(detector_gpu_ptr.get());
return detector_gpu.net.h;
}
YOLODLL_API int Detector::get_net_color_depth() const {
detector_gpu_t &detector_gpu = *static_cast<detector_gpu_t *>(detector_gpu_ptr.get());
return detector_gpu.net.c;
}
YOLODLL_API std::vector<bbox_t> Detector::detect(std::string image_filename, float thresh, bool use_mean)

@ -57,6 +57,7 @@ public:
static YOLODLL_API void free_image(image_t m);
YOLODLL_API int get_net_width() const;
YOLODLL_API int get_net_height() const;
YOLODLL_API int get_net_color_depth() const;
YOLODLL_API std::vector<bbox_t> tracking_id(std::vector<bbox_t> cur_bbox_vec, bool const change_history = true,
int const frames_story = 10, int const max_dist = 150);

Loading…
Cancel
Save