Long caption issue

[hyq217]

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]

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