AlexeyAB
2017-07-24 37921caad3c9a09bcdaba56453e35ab12dabb30f
Fix that allows speedup usage of Yolo-DLL
2 files modified
15 ■■■■ changed files
src/yolo_console_dll.cpp 1 ●●●● patch | view | raw | blame | history
src/yolo_v2_class.hpp 14 ●●●● patch | view | raw | blame | history
src/yolo_console_dll.cpp
@@ -3,6 +3,7 @@
#include <string>
#include <vector>
#include <fstream>
#include <thread>
#define OPENCV
src/yolo_v2_class.hpp
@@ -51,20 +51,20 @@
    {
        if(mat.data == NULL)
            throw std::runtime_error("file not found");
        std::shared_ptr<image_t> image_ptr(new image_t, [](image_t *img) { free_image(*img); delete img; });
        *image_ptr = mat_to_image(mat);
        auto image_ptr = mat_to_image(mat);
        return detect(*image_ptr, thresh);
    }
private:
    static image_t mat_to_image(cv::Mat img)
    static std::shared_ptr<image_t> mat_to_image(cv::Mat img)
    {
        std::shared_ptr<image_t> image_ptr(new image_t, [](image_t *img) { free_image(*img); delete img; });
        std::shared_ptr<IplImage> ipl_small = std::make_shared<IplImage>(img);
        image_t im_small = ipl_to_image(ipl_small.get());
        rgbgr_image(im_small);
        return im_small;
        *image_ptr = ipl_to_image(ipl_small.get());
        rgbgr_image(*image_ptr);
        return image_ptr;
    }
private:
    static image_t ipl_to_image(IplImage* src)
    {
        unsigned char *data = (unsigned char *)src->imageData;