amazon-science / polygon-transformer

Apache License 2.0
128 stars 8 forks source link

Generation of multi-object sequence label #9

Closed big-feather closed 1 year ago

big-feather commented 1 year ago

Dear Author, I would like to know how to generate sequence labels with intervals when there are multiple objects in the mask or a single object with multiple disjointed areas?If it's convenient, could you please point out which part of the code implements this?

joellliu commented 1 year ago

Hi, we currently do support the scenario where a single object has multiple disjointed areas, and the polygons for each area are concatenated with the separator token in between. https://github.com/amazon-science/polygon-transformer/blob/52b83d8eaec7893119ae9885d5e33b5f9f95c83a/data/poly_utils.py#L21

big-feather commented 1 year ago

Thanks for your quick reply!