gpt4 book ai didi

python - 摆脱 Tensorflow Object Detection API 中不同类别的重叠边界框

转载 作者:太空宇宙 更新时间:2023-11-04 04:44:22 25 4
gpt4 key购买 nike

我正在使用 Tensorflow 对象检测 API 来训练我自己的车辆检测器。当我使用 Object detection tutorial 测试我的模型时,我发现在某些情况下,卡车被检测为汽车和卡车,周围有两个重叠的边界框。我只想留下检测分数最高的那个。

我知道对象检测 API 确实会消除重叠的边界框,但它不会针对跨不同类的边界框这样做。有没有办法摆脱重叠的盒子?我可以更改对象检测 API 代码中的任何地方来实现该目标吗?

最佳答案

有两种方法可以消除重叠的边界框。

1、在文件visualization_utils.py中的“visualize_boxes_and_labels_on_image_array”函数中将“min_score_thresh”参数设置得更大;

2、在配置文件中设置非最大抑制阈值

关于python - 摆脱 Tensorflow Object Detection API 中不同类别的重叠边界框,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49951422/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com