Code Request for Instance Matching

Epiphqny / VisTR

[CVPR2021 Oral] End-to-End Video Instance Segmentation with Transformers

https://arxiv.org/abs/2011.14503

Apache License 2.0

738 stars 96 forks source link

Code Request for Instance Matching #65

Closed zyayoung closed 2 years ago

zyayoung commented 2 years ago

In section 4.2, the paper mentions:

if the clip length is less than the video length, instance matching in overlapping frames is used for associating them from different clips.

Code for instance matching is crucial for running VisTR on GPUs with smaller VRAM. Could you kindly provide the corresponding code for this part? :smiley:

Epiphqny commented 2 years ago

Hi @zyayoung, please refer to https://github.com/Epiphqny/VisTR/blob/445c9e4e787a1fb3c959d7e7bb6ecf809bdac155/models/matcher.py#L49.