yolo2检测到的物体位置输出

说明

检测到的物体被box所标记，输出box所在的位置，并存储到txt中。（待完善，如何根据每个图片名字存储不同的txt）

---

进入src/image.c,修改void draw_detections函数。

void draw_detections(image im, int num, float thresh, box *boxes, float **probs, float **masks, char **names, image **alphabet, int classes)
{
    int i;
    char a[100];
    FILE *fp;
    if((fp=fopen("/home/maqy/darknet/ship.txt","w"))==NULL){
        printf("/home/maqy/darknet/ship.txt can't open\n");
        exit(1);
    }

    int shipnum=0;//船的数量
    for(i = 0; i < num; ++i){
        int class = max_index(probs[i], classes);
        float prob = probs[i][class];
        if(prob > thresh){

        ++shipnum;//如果高于阈值，就数目加1

            int width = im.h * .006;

            if(0){
                width = pow(prob, 1./2.)*10+1;
                alphabet = 0;
            }

            //printf("%d %s: %.0f%%\n", i, names[class], prob*100);
            printf("%s: %.0f%% ", names[class], prob*100);
            int offset = class*123457 % classes;
            float red = get_color(2,offset,classes);
            float green = get_color(1,offset,classes);
            float blue = get_color(0,offset,classes);
            float rgb[3];

            //width = prob*20+2;

            rgb[0] = red;
            rgb[1] = green;
            rgb[2] = blue;
            box b = boxes[i];

            int left  = (b.x-b.w/2.)*im.w;//距离图片左边界的值  right-left为box宽度
            int right = (b.x+b.w/2.)*im.w;
            int top   = (b.y-b.h/2.)*im.h;//距离图片上边界的值  bot-top为box高度
            int bot   = (b.y+b.h/2.)*im.h;

            if(left < 0) left = 0;
            if(right > im.w-1) right = im.w-1;
            if(top < 0) top = 0;
            if(bot > im.h-1) bot = im.h-1;

            //maqy-输出box的位置
        printf("%d %d %d %d\n", left, right, top, bot);
        sprintf(a,"%d %d %d %d\n",left,right,top,bot);
        printf("a:%s\n",a);
        fputs(a,fp);

            draw_box_width(im, left, top, right, bot, width, red, green, blue);
            if (alphabet) {
                image label = get_label(alphabet, names[class], (im.h*.03)/10);
                draw_label(im, top + width, left, label, rgb);
                free_image(label);
            }
            if (masks){
                image mask = float_to_image(14, 14, 1, masks[i]);
                image resized_mask = resize_image(mask, b.w*im.w, b.h*im.h);
                image tmask = threshold_image(resized_mask, .5);
                embed_image(tmask, im, left, top);
                free_image(mask);
                free_image(resized_mask);
                free_image(tmask);
            }
        }
    }
    printf("shipNum:%d\n",shipnum);//输出船的数目.
    sprintf(a,"%d\n",shipnum);
    printf("a:%s",a);
    rewind(fp);//回到文件开头
    fputs(a,fp);
    fclose(fp);
}

修改完后需要对darknet重新编译。

修改完后的终端输出：

存储的txt，第一行表示检测出来的总数