search

ashkamath / mdetr

Apache License 2.0
969 stars 125 forks source link

Long caption issue #79

Open hyq217 opened 2 years ago

hyq217 commented 2 years ago

Hi,

I am so appreciated that you have done such a brilliant job. This really attracts me. When I generate segmentation based on a specific dataset. It works fine with those short caption.

But when I try a longer caption like 'the man in the red shirt carrying baseball bats in the corner'.

There is an issue:

assert len(scores) == len(boxes) == len(labels) == len(masks)

AssertionError

I checked the length of 'len(labels)' in function (main.py)plot_inference_segmentation,
len(scores) =5 len(boxes) = 5 len(labels) =0 len(masks)=5,

the 'positive_tokens' is not successfully handled.

Could you please help me with this issue?

Thanks a lot.

BR. Could you please help me with this issue?

Franklin905 commented 6 months ago

Got the same issue. Did you figure out how to fix it?