YOLOv9-S修改

contrl F 定位
定位1 elif m is CBLinear:
修改C2
elif m is CBLinear:
    #yolov9-s 修改
    # c2 = args[0] 原来的
    c2 = [int(x * gw) for x in args[0]] #修改该后的
    c1 = ch[f]

定位2 args = [c1, c2, *args[1:]]

在如上面添加如下代码

#yolov9-s
# --------------------------添加的代码--------------------------- #
if m in (RepNCSPELAN4,):
    args[1] = make_divisible(args[1] * gw, 8)
    args[2] = make_divisible(args[2] * gw, 8)
    args[3] = max(round(args[3] * gd), 1) if n > 1 else n
# --------------------------添加的代码--------------------------- #

参考:训练YOLOv9-S(注意:官方还没有提供YOLOv9-S的网络,我这是根据网络博客进行的步骤,按照0.33、0.50比例调整网络大小,参数量15.60M,计算量67.7GFLOPs)_yolov9s-CSDN博客

  • 3
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
修改YOLOv4-tiny的预测框颜色,可以在源代码中找到相关的函数并进行修改。具体步骤如下: 1. 打开YOLOv4-tiny的源代码文件。 2. 找到预测框绘制函数,一般在文件中名为draw_detections()或者draw_bbox()。 3. 在绘制预测框的代码块中,找到用于绘制框的函数,一般为cv::rectangle()。 4. 在函数中添加一个颜色参数,可以使用OpenCV的Scalar类型指定颜色。 5. 保存修改并重新编译代码。 以下是示例代码: ```cpp void draw_detections(cv::Mat& mat, detection* dets, int num, float thresh, char** names, cv::Mat labels, int classes) { for (int i = 0; i < num; ++i) { if (dets[i].prob[0] > thresh) { int x = (int)dets[i].bbox.x; int y = (int)dets[i].bbox.y; int w = (int)dets[i].bbox.w; int h = (int)dets[i].bbox.h; int label = max_index(dets[i].prob, classes); float prob = dets[i].prob[label]; cv::Scalar color; if (label == 0) // 如果是第一类目标,使用蓝色框 color = cv::Scalar(255, 0, 0); else if (label == 1) // 如果是第二类目标,使用绿色框 color = cv::Scalar(0, 255, 0); else // 其他目标使用红色框 color = cv::Scalar(0, 0, 255); cv::rectangle(mat, cv::Rect(x, y, w, h), color, 2); // 绘制类别标签 char label_str[50]; sprintf(label_str, "%s: %.2f", names[label], prob); int baseline; cv::Size label_size = cv::getTextSize(label_str, cv::FONT_HERSHEY_SIMPLEX, 0.5, 1, &baseline); cv::putText(mat, label_str, cv::Point(x, y - label_size.height), cv::FONT_HERSHEY_SIMPLEX, 0.5, color, 1, cv::LINE_AA); } } } ``` 在上面的代码中,我们根据类别修改了不同颜色的预测框。您可以根据自己的需求修改颜色和绘制方式。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值