YOLOv6量化后仅能预测一类

[zhujiajian98]

我采用了你们在YOLOv6官方仓库的量化代码[(https://github.com/meituan/YOLOv6/tree/main/tools/quantization/ppq)]，但是在完成所有的流程后，仅能预测一类出来，我猜测这跟nms中你们给的nms_forward_function函数，因为它内部仅定义了全0的向量，这导致了无法预测出出了类别0 (person)以外的类别，不知道你们能否给出有效的nm_forward_function函数呢？非常感谢！

[ZhangZhiPku]

貌似..你并不需要那个nms函数，应该是其他问题导致了上述情况，你可以提供yolo6.onnx给我吗？

[zhujiajian98]

可以的，请问您的邮箱是多少？我直接发给您。

[zhujiajian98]

我现在已经可以成功量化了，但是就是推理出来的只有针对类别0的框，其他类别是无法推理的，现在非常疑惑这个点。