LiWentomng / BoxInstSeg

A toolbox for box-supervised instance segmentation.
Apache License 2.0
407 stars 38 forks source link

box2mask_c导出onnx报错 #26

Open ZHIZIHUABU opened 11 months ago

ZHIZIHUABU commented 11 months ago

我使用box2mask_r50_lsj_8x2_50e_duanmian.py配置文件训练模型,模型训练完成后导出onnx报错,在debug过程中发现会跳转到maskformer.py中,但是跳转到其中的self.onnx_export()函数时,函数为空,raise NotImplementedError。 请问是我使用错误,还是代码并未实现呢?single_stage.py中是有onnx_export.py实现的

LiWentomng commented 10 months ago

@ZHIZIHUABU Sorry for the late reply. 该代码中暂未实现onnx, 如需要此功能,请参考mmdetection V2.25.0版本进行实现。

StephenZhao1 commented 1 month ago

我使用box2mask_r50_lsj_8x2_50e_duanmian.py配置文件训练模型,模型训练完成后导出onnx报错,在debug过程中发现会跳转到maskformer.py中,但是跳转到其中的self.onnx_export()函数时,函数为空,raise NotImplementedError。 请问是我使用错误,还是代码并未实现呢?single_stage.py中是有onnx_export.py实现的

请问你转成功了吗?我们这边也需要做部署,做tensorrt或者openvino的