替换cython_bbox库中bbox_ious

说明一下问题:目标追踪代码里往往用到cython_bbox中的bbox_ious。但是该库需要用到 VC++,按照有些麻烦。于是采用直接替换该方法,用纯代码实现,无需调用库。

File “src\cython_bbox.pyx”, line 17, in cython_bbox.bbox_overlaps
ValueError: Buffer dtype mismatch, expected ‘DTYPE_t’ but got ‘float’

代码来源:
PaddleDetection-release-2.7\build\lib\ppdet\data\transform\atss_assigner.py

def bbox_ious(bboxes1, bboxes2, mode='iou', is_aligned=False, eps=1e-6):"""Calculate overlap between two set of bboxes.If ``is_aligned `` is ``False``, then calculate the overlaps between eachbbox of bboxes1 and bboxes2, otherwise the overlaps between each alignedpair of bboxes1 and bboxes2.Args:bboxes1 (Tensor): shape (B, m, 4) in <x1, y1, x2, y2> format or empty.bboxes2 (Tensor): shape (B, n, 4) in <x1, y1, x2, y2> format or empty.B indicates the batch dim, in shape (B1, B2, ..., Bn).If ``is_aligned `` is ``True``, then m and n must be equal.mode (str): "iou" (intersection over union) or "iof" (intersection overforeground).is_aligned (bool, optional): If True, then m and n must be equal.Default False.eps (float, optional): A value added to the denominator for numericalstability. Default 1e-6.Returns:Tensor: shape (m, n) if ``is_aligned `` is False else shape (m,)"""assert mode in ['iou', 'iof', 'giou', 'diou'], 'Unsupported mode {}'.format(mode)# Either the boxes are empty or the length of boxes's last dimenstion is 4assert (bboxes1.shape[-1] == 4 or bboxes1.shape[0] == 0)assert (bboxes2.shape[-1] == 4 or bboxes2.shape[0] == 0)# Batch dim must be the same# Batch dim: (B1, B2, ... Bn)assert bboxes1.shape[:-2] == bboxes2.shape[:-2]batch_shape = bboxes1.shape[:-2]rows = bboxes1.shape[-2] if bboxes1.shape[0] > 0 else 0cols = bboxes2.shape[-2] if bboxes2.shape[0] > 0 else 0if is_aligned:assert rows == colsif rows * cols == 0:if is_aligned:return np.random.random(batch_shape + (rows, ))else:return np.random.random(batch_shape + (rows, cols))area1 = (bboxes1[..., 2] - bboxes1[..., 0]) * (bboxes1[..., 3] - bboxes1[..., 1])area2 = (bboxes2[..., 2] - bboxes2[..., 0]) * (bboxes2[..., 3] - bboxes2[..., 1])if is_aligned:lt = np.maximum(bboxes1[..., :2], bboxes2[..., :2])  # [B, rows, 2]rb = np.minimum(bboxes1[..., 2:], bboxes2[..., 2:])  # [B, rows, 2]wh = (rb - lt).clip(min=0)  # [B, rows, 2]overlap = wh[..., 0] * wh[..., 1]if mode in ['iou', 'giou']:union = area1 + area2 - overlapelse:union = area1if mode == 'giou':enclosed_lt = np.minimum(bboxes1[..., :2], bboxes2[..., :2])enclosed_rb = np.maximum(bboxes1[..., 2:], bboxes2[..., 2:])if mode == 'diou':enclosed_lt = np.minimum(bboxes1[..., :2], bboxes2[..., :2])enclosed_rb = np.maximum(bboxes1[..., 2:], bboxes2[..., 2:])b1_x1, b1_y1 = bboxes1[..., 0], bboxes1[..., 1]b1_x2, b1_y2 = bboxes1[..., 2], bboxes1[..., 3]b2_x1, b2_y1 = bboxes2[..., 0], bboxes2[..., 1]b2_x2, b2_y2 = bboxes2[..., 2], bboxes2[..., 3]else:lt = np.maximum(bboxes1[..., :, None, :2],bboxes2[..., None, :, :2])  # [B, rows, cols, 2]rb = np.minimum(bboxes1[..., :, None, 2:],bboxes2[..., None, :, 2:])  # [B, rows, cols, 2]wh = (rb - lt).clip(min=0)  # [B, rows, cols, 2]overlap = wh[..., 0] * wh[..., 1]if mode in ['iou', 'giou']:union = area1[..., None] + area2[..., None, :] - overlapelse:union = area1[..., None]if mode == 'giou':enclosed_lt = np.minimum(bboxes1[..., :, None, :2],bboxes2[..., None, :, :2])enclosed_rb = np.maximum(bboxes1[..., :, None, 2:],bboxes2[..., None, :, 2:])if mode == 'diou':enclosed_lt = np.minimum(bboxes1[..., :, None, :2],bboxes2[..., None, :, :2])enclosed_rb = np.maximum(bboxes1[..., :, None, 2:],bboxes2[..., None, :, 2:])b1_x1, b1_y1 = bboxes1[..., :, None, 0], bboxes1[..., :, None, 1]b1_x2, b1_y2 = bboxes1[..., :, None, 2], bboxes1[..., :, None, 3]b2_x1, b2_y1 = bboxes2[..., None, :, 0], bboxes2[..., None, :, 1]b2_x2, b2_y2 = bboxes2[..., None, :, 2], bboxes2[..., None, :, 3]eps = np.array([eps])union = np.maximum(union, eps)ious = overlap / unionif mode in ['iou', 'iof']:return ious# calculate giousif mode in ['giou']:enclose_wh = (enclosed_rb - enclosed_lt).clip(min=0)enclose_area = enclose_wh[..., 0] * enclose_wh[..., 1]enclose_area = np.maximum(enclose_area, eps)gious = ious - (enclose_area - union) / enclose_areareturn giousif mode in ['diou']:left = ((b2_x1 + b2_x2) - (b1_x1 + b1_x2))**2 / 4right = ((b2_y1 + b2_y2) - (b1_y1 + b1_y2))**2 / 4rho2 = left + rightenclose_wh = (enclosed_rb - enclosed_lt).clip(min=0)enclose_c = enclose_wh[..., 0]**2 + enclose_wh[..., 1]**2enclose_c = np.maximum(enclose_c, eps)dious = ious - rho2 / enclose_creturn dious

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/bicheng/53659.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

深度学习-01 Pytorch

torchvision是一个用于计算机视觉任务的Python包&#xff0c;它是PyTorch的一个扩展库。它提供了一些流行的数据集、模型架构和图像转换函数&#xff0c;以方便用户进行计算机视觉任务的开发和研究。 1.torchvision中包含了许多常用的计算机视觉数据集&#xff0c;如MNIST、CIF…

【AcWing】861. 二分图的最大匹配(匈牙利算法)

匈牙利算法&#xff0c;他可以在比较快的时间复杂度之内告诉我们左边和右边成功匹配的最大数是多少 匹配指的是边的数量&#xff0c;成功的匹配指的是两个未被使用的点之间存在一条边(就不存在两条边共用了一个点的)。 匈牙利算法可以返回成功匹配的最大匹配数是多少。 #incl…

四、搭建网站服务器超详细步骤——解决宝塔界面无法登录问题

前言 本篇博客是搭建网站服务器的第四期&#xff0c;也到了中间的一节 先分享一下我在搭建网站时的个人感受&#xff0c;我在这个环节卡住了很久 后来突然醒悟了&#xff0c;然后成功进入了宝塔界面 现在就来分享一下&#xff0c;我所遇到的问题 小伙伴们坐好了 …

MySQL8 windows archive 安装

MySQL8超详细安装教程&#xff08;MySQL8安装图文教程&#xff09;_MySQL8安装包-CSDN博客 【补充&#xff1a;】 1.mysqld --install mysql // mysql是本次安装的服务命名&#xff0c;也可以叫做mysql8 etc. 2.手动卸载 2.1.net stop mysql 2.2.删除datadir // 即my.ini配置…

项目实战 ---- 商用落地视频搜索系统(9)---UI与上层service的交互优化

目录 背景 第一次优化(UI优化) 优化前UI 优化方向与问题 代码 修改效果 第二次优化(整合优化) 优化方向与问题 代码 优化效果 第三次优化(js层优化) 优化方向与问题 代码 优化效果 第四次优化(UI逻辑再优化) 优化方向与问题 代码 优化效果 第五次优化(纯U…

通信工程学习:什么是FMC固定移动融合

FMC&#xff1a;固定移动融合 FMC固定移动融合&#xff0c;即Fixed Mobile Convergence&#xff08;固定移动网络融合&#xff09;&#xff0c;是指通过固定网络与移动网络之间的融通、合作&#xff0c;实现全业务及融合业务的经营。这一技术旨在打破传统固定网络和移动网络之间…

使用 Parallel 类进行多线程编码(上)

用 C# 进行多线程编程有很多方式&#xff0c;比如使用 Thread 对象开启一个新线程&#xff0c;但这已经是一种落后的写法了&#xff0c;现在推荐的写法是使用 Parallel 类&#xff0c;它可以让我们像写传统代码一样编写多线程的程序&#xff0c;Parallel 类有三个常用的方法如下…

C#笔记8 线程是什么?多线程怎么实现和操作?

这和前面的学习内容可能有点不太连贯&#xff0c;但是呢我们一般来说的学习就是遇到什么困难就去学习什么&#xff0c;这也是为什么看那些循序渐进的教程虽然学的很饱满&#xff0c;但是我们有时会学了前面忘记了后面&#xff0c;或者对某个板块理解不深&#xff0c;乃至于写代…

VS Code 中提升编程效率的功能及使用方法

Visual Studio Code&#xff08;简称 VS Code&#xff09;是一款功能强大且灵活的源代码编辑器&#xff0c;支持多种编程语言&#xff0c;并提供了丰富的插件生态系统。对于开发者来说&#xff0c;掌握一些提升编程效率的功能是非常重要的。本文将介绍 VS Code 中一些能够显著提…

大学物理实验-杨氏双缝干涉实验

一、实验目的 1.理解杨氏双缝干涉现象的基本原理&#xff1b; 2.掌握用光具座调整杨氏双缝干涉装置的光路&#xff1b; 3.用CCD观察杨氏双缝干涉的实验现象&#xff1b; 4.学会用干涉法测量未知光波的波长。 二、仪器用具 FB760-6型光学实验仪多功能光学片 三、实验原理 1.波…

智能提取:OfficeImagesExtractor让文档图片提取更简单

“科技是国之利器&#xff0c;也是民之福祉。” 在数字化办公日益普及的今天&#xff0c;我们对文档处理的需求也在不断增长。尤其是对于Office文档中的图片、视频和音频等多媒体内容的提取&#xff0c;传统的方法是繁琐且效率低下的。在这样的背景下&#xff0c;一款能够高效、…

计算机毕业设计选题推荐-产品委托配送系统-Java/Python项目实战

✨作者主页&#xff1a;IT毕设梦工厂✨ 个人简介&#xff1a;曾从事计算机专业培训教学&#xff0c;擅长Java、Python、微信小程序、Golang、安卓Android等项目实战。接项目定制开发、代码讲解、答辩教学、文档编写、降重等。 ☑文末获取源码☑ 精彩专栏推荐⬇⬇⬇ Java项目 Py…

视频监控管理平台LntonAIServer视频智能分析噪声检测应用场景

在视频监控系统中&#xff0c;噪声问题常常影响到视频画面的清晰度和可用性。噪声可能由多种因素引起&#xff0c;包括但不限于低光环境、摄像机传感器灵敏度过高、编码压缩失真等。LntonAIServer通过引入噪声检测功能&#xff0c;旨在帮助用户及时发现并解决视频流中的噪声问题…

原点安全荣获“AutoSec Awards 安全之星”优秀汽车数据安全合规方案奖

9月3日&#xff0c;「AutoSec 2024第八届中国汽车网络安全周暨第五届智能汽车数据安全展」在上海盛大开幕。本届大会由谈思实验室和谈思汽车主办、上海市车联网协会联合主办&#xff0c;以汽车“网络数据安全、软件安全、功能安全”为主题&#xff0c;汇聚了国内外的技术专家、…

公共组件和业务组件的区别

公共基础组件是构成用户界面的最小单位&#xff0c;如按钮、输入框、卡片等&#xff0c;封装基础组件&#xff0c;应考虑以下几点&#xff1a; 通用性&#xff1a;确保组件能够在不同场景下复用可配置性&#xff1a;提供灵活的属性配置&#xff0c;以适应不同的设计需求无状态…

第四次北漂----挣个独立游戏的素材钱

第四次北漂&#xff0c;在智联招聘上&#xff0c;有个小公司主动和我联系。面试了下&#xff0c;决定入职了&#xff0c;osg/osgearth的。月薪两万一。 大跌眼镜的是&#xff0c;我入职后&#xff0c;第一天的工作内容就是接手他的工作&#xff0c;三天后他就离职了。 我之所以…

sysbench下载与交叉编译

简介 sysbench是一个开源的多线程性能测试工具&#xff0c;广泛用于评估系统、数据库和硬件的性能。它支持多种测试模式&#xff0c;如 CPU、内存、磁盘 I/O 和数据库基准测试等。 下载 git clone https://gitee.com/cn-loongson/sysbench.git 这里使用国内的源&#xff0c…

宏观学习笔记:GDP分析(二)

GDP分析&#xff08;一&#xff09;主要是介绍GDP相关的定义以及核算逻辑&#xff0c;本节主要介绍GDP的分析思路。GDP分析主要是2种方法&#xff1a;总量分析和结构分析。 1. 总量分析 1.1 数值选择 一般情况下&#xff0c;分析的对象都是 官方公布的GDP当季值。 1.2 趋势规…

全国-住宅区AOI数据

数据量级&#xff1a;54万&#xff0c;更新时间&#xff1a;2024年3月 覆盖字段&#xff1a; 名称&#xff0c;地址&#xff0c;经纬度&#xff0c;一级分类&#xff0c;二级分类&#xff0c;三级分类&#xff0c;默认图片&#xff0c;AOI围栏 数据来源于&#xff1a;魔行观察…

MySQL——视图(二)视图管理(7)删除视图

当视图不再需要时&#xff0c;可以将其删除&#xff0c;删除视图时&#xff0c;只能删除视图的定义&#xff0c;不会删除数据。删除一个或多个视图可以使用 DROP VIEW 语句&#xff0c;删除视图的基本语法格式如下所示&#xff1a; DROP VIEWIIE EXISTS]view_name [,view_name…