The 物体检测笔记本演示,如何models预训练于COCO数据集可用于检测测试图像上的物体。但是,笔记本中的模型会返回 COCO 集中所有类别的检测到对象的框。如何使用代码仅返回一种类别的对象的框? IE。我怎样才能获得模型确定的对象的盒子,例如人?
我刚刚自己实施了该解决方案。
检查def filter_boxes
笔记本中的功能https://github.com/fera0013/TrafficLightDetection/blob/master/Traffic-Light-Detection.ipynb,查看如何按类别过滤框。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)