Alexey
2018-03-01 17b22d7ce77cf37ac7d39b5ea3be76716c10cfdf
src/image.c
@@ -13,8 +13,14 @@
#ifdef OPENCV
#include "opencv2/highgui/highgui_c.h"
#include "opencv2/imgproc/imgproc_c.h"
#include "opencv2/core/version.hpp"
#include "http_stream.h"
#ifndef CV_VERSION_EPOCH
#include "opencv2/videoio/videoio_c.h"
#include "opencv2/imgcodecs/imgcodecs_c.h"
#include "http_stream.h"
#endif
#endif
int windows = 0;
@@ -186,6 +192,14 @@
        float prob = probs[i][class];
        if(prob > thresh){
         //// for comparison with OpenCV version of DNN Darknet Yolo v2
         //printf("\n %f, %f, %f, %f, ", boxes[i].x, boxes[i].y, boxes[i].w, boxes[i].h);
         // int k;
         //for (k = 0; k < classes; ++k) {
         // printf("%f, ", probs[i][k]);
         //}
         //printf("\n");
            int width = im.h * .012;
            if(0){
@@ -285,13 +299,13 @@
         color.val[2] = blue * 256;
         cvRectangle(show_img, pt1, pt2, color, width, 8, 0);
         //printf("left=%d, right=%d, top=%d, bottom=%d, obj_id=%d, obj=%s \n", left, right, top, bot, class, names[class]);
         cvRectangle(show_img, pt_text_bg1, pt_text_bg2, color, width, 8, 0);
         cvRectangle(show_img, pt_text_bg1, pt_text_bg2, color, CV_FILLED, 8, 0);   // filled
         CvScalar black_color;
         black_color.val[0] = 0;
         CvFont font;
         cvInitFont(&font, CV_FONT_HERSHEY_COMPLEX, font_size, font_size, 0, font_size*3, 8);
         cvInitFont(&font, CV_FONT_HERSHEY_SIMPLEX, font_size, font_size, 0, font_size * 3, 8);
         cvPutText(show_img, names[class], pt_text, &font, black_color);
      }
   }
@@ -514,7 +528,7 @@
}
void show_image_cv_ipl(IplImage *disp, const char *name)
void show_image_cv_ipl(IplImage *disp, const char *name, CvVideoWriter *output_video_writer, int http_stream_port)
{
   if (disp == NULL) return;
   char buff[256];
@@ -526,24 +540,18 @@
   cvShowImage(buff, disp);
   {
      CvSize size;
      {
         size.width = disp->width, size.height = disp->height;
      }
      static CvVideoWriter* output_video = NULL;    // cv::VideoWriter output_video;
      if (output_video == NULL)
      {
         //printf("\n SRC output_video = %p \n", output_video);
         const char* output_name = "test_dnn_out.avi";
         //output_video = cvCreateVideoWriter(output_name, CV_FOURCC('H', '2', '6', '4'), 25, size, 1);
         output_video = cvCreateVideoWriter(output_name, CV_FOURCC('D', 'I', 'V', 'X'), 25, size, 1);
         //output_video = cvCreateVideoWriter(output_name, CV_FOURCC('M', 'J', 'P', 'G'), 25, size, 1);
         //printf("\n cvCreateVideoWriter, DST output_video = %p  \n", output_video);
      }
   // http mjpeg stream: http://localhost:8090
   // use URL with the port number stated in your command line instead of 8090
   if (http_stream_port > 0) {
      //int port = 8090;
      int port = http_stream_port;
      int timeout = 200;
      int jpeg_quality = 30;  // 1 - 100
      send_mjpeg(disp, port, timeout, jpeg_quality);
   }
      cvWriteFrame(output_video, disp);   // comment this line to improve FPS !!!
   if(output_video_writer) {
      cvWriteFrame(output_video_writer, disp);  // comment this line to improve FPS !!!
      printf("\n cvWriteFrame \n");
   }
@@ -626,8 +634,8 @@
   *in_img = cvCreateImage(cvSize(src->width, src->height), IPL_DEPTH_8U, 3);
   cvResize(src, *in_img, CV_INTER_LINEAR);
   cvResize(src, new_img, CV_INTER_LINEAR);
   src = new_img;
   image im = ipl_to_image(src);
   image im = ipl_to_image(new_img);
   cvReleaseImage(&new_img);
   rgbgr_image(im);
   return im;
}
@@ -877,6 +885,51 @@
#endif
}
void fill_image(image m, float s)
{
   int i;
   for (i = 0; i < m.h*m.w*m.c; ++i) m.data[i] = s;
}
void letterbox_image_into(image im, int w, int h, image boxed)
{
   int new_w = im.w;
   int new_h = im.h;
   if (((float)w / im.w) < ((float)h / im.h)) {
      new_w = w;
      new_h = (im.h * w) / im.w;
   }
   else {
      new_h = h;
      new_w = (im.w * h) / im.h;
   }
   image resized = resize_image(im, new_w, new_h);
   embed_image(resized, boxed, (w - new_w) / 2, (h - new_h) / 2);
   free_image(resized);
}
image letterbox_image(image im, int w, int h)
{
   int new_w = im.w;
   int new_h = im.h;
   if (((float)w / im.w) < ((float)h / im.h)) {
      new_w = w;
      new_h = (im.h * w) / im.w;
   }
   else {
      new_h = h;
      new_w = (im.w * h) / im.h;
   }
   image resized = resize_image(im, new_w, new_h);
   image boxed = make_image(w, h, im.c);
   fill_image(boxed, .5);
   //int i;
   //for(i = 0; i < boxed.w*boxed.h*boxed.c; ++i) boxed.data[i] = 0;
   embed_image(resized, boxed, (w - new_w) / 2, (h - new_h) / 2);
   free_image(resized);
   return boxed;
}
image resize_max(image im, int max)
{
    int w = im.w;
@@ -1151,7 +1204,7 @@
void random_distort_image(image im, float hue, float saturation, float exposure)
{
    float dhue = rand_uniform(-hue, hue);
    float dhue = rand_uniform_strong(-hue, hue);
    float dsat = rand_scale(saturation);
    float dexp = rand_scale(exposure);
    distort_image(im, dhue, dsat, dexp);
@@ -1303,9 +1356,16 @@
image load_image(char *filename, int w, int h, int c)
{
#ifdef OPENCV
    image out = load_image_cv(filename, c);
#ifndef CV_VERSION_EPOCH
   //image out = load_image_stb(filename, c);   // OpenCV 3.x
   image out = load_image_cv(filename, c);
#else
    image out = load_image_stb(filename, c);
   image out = load_image_cv(filename, c);      // OpenCV 2.4.x
#endif
#else
    image out = load_image_stb(filename, c); // without OpenCV
#endif
    if((h && w) && (h != out.h || w != out.w)){