使用huggingface transformers owlvit进行one shot目标检测时置信度score太高的问题

lanlinbuaa

已于 2024-01-23 17:14:27 修改

阅读量615

点赞数 8

文章标签：目标检测人工智能计算机视觉

于 2024-01-09 17:53:09 首次发布

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/lanlinjnc/article/details/135486018

版权

文章探讨了在使用HuggingFacetransformers的owlvit模型进行oneshot目标检测时，尽管引导图与待检测图存在显著差异，但模型的置信度分数仍然过高。主要原因是后处理函数对置信度得分进行了再映射，导致结果不准确。

摘要由CSDN通过智能技术生成

问题

使用 huggingface 的 transformers 库中的 owlvit 模型进行 one shot 目标检测时，引导图与待检测图差距巨大，但最终置信度score依旧太高

实验

实验1

待检测图像1

请添加图片描述

引导图1

请添加图片描述

检测结果1

在这里插入图片描述

实验2

待检测图像2

在这里插入图片描述

引导图1

在这里插入图片描述

检测结果1

在这里插入图片描述

原因

transformers中的后处理函数中对置信度scores进行了再映射，即将预测的最大值又映射为1.00，后处理函数为OwlViTProcessor.post_process_image_guided_detection()，具体调用的是image_processing_owlvit.py文件中的post_process_image_guided_detection函数，如下图所示

在这里插入图片描述

关注

8
点赞
踩
9

收藏

觉得还不错? 一键收藏
0
评论
使用huggingface transformers owlvit进行one shot目标检测时置信度score太高的问题

使用 huggingface 的 transformers 库中的 owlvit 模型进行 one shot 目标检测时，引导图与待检测图差距巨大，但最终置信度score依旧太高。
复制链接

扫一扫

评论

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

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