seoungwugoh / STM

Video Object Segmentation using Space-Time Memory Networks
405 stars 81 forks source link

A question in the single object segmentation #34

Closed HengyiWang closed 3 years ago

HengyiWang commented 3 years ago

Hey, I think in the dataset.py, line 79 https://github.com/seoungwugoh/STM/blob/905f11492a6692dd0d0fa395881a8ec09b211a36/dataset.py#L79 It may be N_masks <= 255 instead of N_masks < 255 ?