AlexeyAB
2018-03-20 fe4f01b1d8cce9762e47aaf42b6e17a3427a144b
src/image.c
@@ -14,12 +14,12 @@
#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
#include "http_stream.h"
#endif
int windows = 0;
@@ -207,7 +207,6 @@
                alphabet = 0;
            }
            printf("%s: %.0f%%\n", names[class_id], prob*100);
            int offset = class_id*123457 % classes;
            float red = get_color(2,offset,classes);
            float green = get_color(1,offset,classes);
@@ -230,7 +229,12 @@
            if(right > im.w-1) right = im.w-1;
            if(top < 0) top = 0;
            if(bot > im.h-1) bot = im.h-1;
         printf("%s: %.0f%%", names[class_id], prob * 100);
         //printf(" - id: %d, x_center: %d, y_center: %d, width: %d, height: %d",
         // class_id, (right + left) / 2, (bot - top) / 2, right - left, bot - top);
         printf("\n");
            draw_box_width(im, left, top, right, bot, width, red, green, blue);
            if (alphabet) {
                image label = get_label(alphabet, names[class_id], (im.h*.03)/10);
@@ -348,6 +352,7 @@
   }
   cvPutText(img, "Iteration number", cvPoint(draw_size / 2, img_size - 10), &font, CV_RGB(0, 0, 0));
   cvPutText(img, "Press 's' to save: chart.jpg", cvPoint(5, img_size - 10), &font, CV_RGB(0, 0, 0));
   printf(" If error occurs - run training with flag: -dont_show \n");
   cvNamedWindow("average loss", CV_WINDOW_NORMAL);
   cvMoveWindow("average loss", 0, 0);
   cvResizeWindow("average loss", img_size, img_size);
@@ -678,9 +683,12 @@
    return im;
}
image get_image_from_stream_resize(CvCapture *cap, int w, int h, IplImage** in_img)
image get_image_from_stream_resize(CvCapture *cap, int w, int h, IplImage** in_img, int use_webcam)
{
   IplImage* src = cvQueryFrame(cap);
   IplImage* src;
   if (use_webcam) src = get_webcam_frame(cap);
   else src = cvQueryFrame(cap);
   if (!src) 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);