Merge branch 'master' of https://github.com/AlexeyAB/darknet into Fix_get_c...

IlyaOvodov

2018-06-04 f0abcfa02b2094396f955c743f7f11fcdb2e3d13

 src/image.c

@@ -957,7 +957,7 @@
{
    IplImage* src = 0;
    int flag = -1;
    if (channels == 0) flag = -1;
    if (channels == 0) flag = 1;
    else if (channels == 1) flag = 0;
    else if (channels == 3) flag = 1;
    else {
@@ -975,6 +975,7 @@
    }
    image out = ipl_to_image(src);
    cvReleaseImage(&src);
   if (out.c > 1)
    rgbgr_image(out);
    return out;
}
@@ -1010,8 +1011,9 @@
   return im;
}

image get_image_from_stream_resize(CvCapture *cap, int w, int h, IplImage** in_img, int cpp_video_capture)
image get_image_from_stream_resize(CvCapture *cap, int w, int h, int c, IplImage** in_img, int cpp_video_capture, int dont_close)
{
   c = c ? c : 3;
   IplImage* src;
   if (cpp_video_capture) {
      static int once = 1;
@@ -1027,15 +1029,23 @@
   }
   else src = cvQueryFrame(cap);

   if (!src) return make_empty_image(0, 0, 0);
   if (src->width < 1 || src->height < 1 || src->nChannels < 1) return make_empty_image(0, 0, 0);
   IplImage* new_img = cvCreateImage(cvSize(w, h), IPL_DEPTH_8U, 3);
   *in_img = cvCreateImage(cvSize(src->width, src->height), IPL_DEPTH_8U, 3);
   if (!src) { 
      if (dont_close) src = cvCreateImage(cvSize(416, 416), IPL_DEPTH_8U, c);
      else return make_empty_image(0, 0, 0); 
   }
   if (src->width < 1 || src->height < 1 || src->nChannels < 1) {
      if (cpp_video_capture) cvReleaseImage(&src);
      if (dont_close) src = cvCreateImage(cvSize(416, 416), IPL_DEPTH_8U, c);
      else return make_empty_image(0, 0, 0);
   }
   IplImage* new_img = cvCreateImage(cvSize(w, h), IPL_DEPTH_8U, c);
   *in_img = cvCreateImage(cvSize(src->width, src->height), IPL_DEPTH_8U, c);
   cvResize(src, *in_img, CV_INTER_LINEAR);
   cvResize(src, new_img, CV_INTER_LINEAR);
   image im = ipl_to_image(new_img);
   cvReleaseImage(&new_img);
   if (cpp_video_capture) cvReleaseImage(&src);
   if (c>1)
   rgbgr_image(im);
   return im;
}
@@ -1589,6 +1599,8 @@

void distort_image(image im, float hue, float sat, float val)
{
   if (im.c >= 3)
   {
    rgb_to_hsv(im);
    scale_image_channel(im, 1, sat);
    scale_image_channel(im, 2, val);
@@ -1599,6 +1611,11 @@
        if (im.data[i] < 0) im.data[i] += 1;
    }
    hsv_to_rgb(im);
   }
   else
   {
      scale_image_channel(im, 0, val);
   }
    constrain_image(im);
}