smoke测试脚本中vis.py无3D框显示

[happyday-lkj]

如题，按照smoke的步骤进行模型转换，然后进行vis.py测试，选用kitti的图片，模型可以推理，但是保存的结果没有框？

[happyday-lkj]

[Yutong-gannis]

@happyday-lkj 调整一下置信度会不会好点

[happyday-lkj]

调整为0.2，也显示不对，在上图的最左侧中间会有一个小框，但是这明显是不对的

[Yutong-gannis]

@happyday-lkj 那你应该没训练，作者只给了预训练权重

[happyday-lkj]

@happyday-lkj 那你应该没训练，作者只给了预训练权重

是的，我是直接用作者给出的权重，这个权重不能用吗？

[Yutong-gannis]

@happyday-lkj 检测不出来的，只有自己训练

[happyday-lkj]

@happyday-lkj 检测不出来的，只有自己训练

readMe贴出来的模型不也是使用kitti数据训练的吗？需要我再使用这个预训练权重再重新训练一下kitti数据集吗？

[Yutong-gannis]

@happyday-lkj 是的，训练后才看得到效果

[happyday-lkj]

@happyday-lkj 是的，训练后才看得到效果
好的，我后面训练一下在看下结果

[FlyingQianMM]

如果您使用的我们提供的预训练模型，且只在kitti数据集上做推理可视化，一般会有预测结果的，如前面所说，可以调整可视化的置信度。SMOKE因为是只用到单目输入，所以精度比较低，可以换几张图片试试。具体的可视化参考 https://github.com/PaddlePaddle/Paddle3D/tree/release/1.0/docs/models/smoke#python%E9%83%A8%E7%BD%B2
如果您是在自己的数据集上做测试，还是要重新训练下才会有比较好的效果

[Ywenjun123]

@FlyingQianMM
我使用python vis.py --model_file /path/to/smoke.pdmodel --params_file /path/to/smoke.pdiparams --image /path/to/image
为什么会
cv2.error: OpenCV(4.5.5) 👎 error: (-5:Bad argument) in function 'circle'

Overload resolution failed:

• Can't parse 'center'. Sequence item with index 0 has a wrong type
• Can't parse 'center'. Sequence item with index 0 has a wrong type

[Ywenjun123]

@happyday-lkj 你现在可以显示3D框了吗

[nepeplwu]

@Ywenjun123 cv2的报错辛苦给下更全的堆栈

[haheheixi]

@happyday-lkj 我觉得 你犯了和我一样的错误 使用hrnet18模型 但在使用python tools/export.py命令导出模型时使用的默认smoke_dla34_no_dcn_kitti.yml配置文件 我改成configs/smoke/smoke_hrnet18_no_dcn_kitti.yml后 可以输出正确的3d框