mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Fix that allows speedup usage of Yolo-DLL

Browse Source
This commit is contained in:
AlexeyAB
2017-07-24 21:43:44 +03:00
parent d6ea2ef026
commit 37921caad3
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/yolo_v2_class.hpp
View File

@ -51,20 +51,20 @@ public:
{
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;