输出yolo的测试结果,根据坐标裁剪原图并保存

本文介绍如何根据Yolo的测试结果坐标,裁剪原图并保存。主要修改了`src/image.c`中的`draw_detections`函数,添加了保存文件的功能。在`darknet.h`中更新函数定义,并调整`examples/detector.c`中的`test_detector`函数以处理批量图片。通过这些修改,实现了将检测到的前9个框保存到单独图像文件的目标。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

因为项目中的需要,本篇博文实现输出(保存)yolo的测试结果,并测试结果的坐标位置切割原图,并不需要知道每个框的类别,保存了top9。
主要对src/image.c文件中的draw_detections函数做了修改。

//添加了 char *filename ,为了得到当前的图片名。
void draw_detections(image im, int num, float thresh, box *boxes, float **probs, float **masks, char **names, image **alphabet, int classes, char *filename)
{
    printf("num %d\n", num);  
    float rawmax[num];
    int i,j;
    int params[3];
    char savePath[100] = "";
    //为了得到top19的框,将每个框按照最大概率值进行了排序,取了top19. b,c矩阵都是为了得到top9对应的原框的下标,好得到坐标点。
    for(i =0; i<num; i++){
    rawmax[i] = probs[i][0];
    for(j =0;j<classes;j++){
         if(probs[i][j] > rawmax[i]){
          rawmax[i] = probs[i][j];                                     }
    }
    }
    for( i =0;i<num;i++){
      if(rawmax[i] > 0)
       printf("rawmax[ %d]:%f\n",i,rawmax[i]);
    }
   float b[num];
   int c[num];

   for(i =0; i<num; c[i]= 1+ i++){
      b[i] = rawmax[i];
   }
   int x;
   for(i =0;i<num;i++){
      for(x = i,j = x+1; j<num;j++)
       if(b[x]<b[j]) x = j;
       if(x!=j){
        j= b[i];
        b[i] = b[x];
        b[x] = j;
        j = c[i];
        c[i] = c[x];
        c[x] = j;
     }
      }
  //输出top9的坐标,后面的代码注释掉了,因为我只需要根据坐标切割出子
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值