search

raoyongming / DenseCLIP

[CVPR 2022] DenseCLIP: Language-Guided Dense Prediction with Context-Aware Prompting
505 stars 38 forks source link

An error while saving the model in ONNX format #32

Closed Virgilzzz closed 1 year ago

Virgilzzz commented 1 year ago

Have you ever tried saving the model as ONNX for visualization? Why does the trained model report an error missing img_meta when saved as ONNX?

raoyongming commented 1 year ago

Hi, thanks for your interest in our work. We don't have much experience in using ONNX. Our code is based on mmseg and mmdet. I think maybe you can check how to use ONNX for mmseg/mmdet models.