How to get the mask for mots task? about unitrack HOT 4 CLOSED

Hi @JessicaChanzc,
If you noticed, the string starts with "UkU" encodes the object masks. This is called the RLE format, widely used to compactly encode masks with strings, see here.

You can use the cocoapi (link above) to encode masks to RLE strings. And also to decode RLE strings to numpy arrays.

from unitrack.

Thanks for your reply,

It seems that in your test_mots.py , you don't load any weights? but just visualize the tracking results?

I'm really appreciate your work and could you pls tell me how to test on my own dataset?

Best regards

from unitrack.

@JessicaChanzc

• The appearance model loads its weights when initialization, see here and here. When using ImageNet pre-train models, we just load default torchvision weights by setting pretrain=True.
• If you want to test your own dataset, I recommend preparing the dataset in the same format as MOTS (video frames, detections). The output, of course, will be in the same format as the MOTS result text. Another alternative is to combine the segmentation model with UniTrack, which may need some time coding but not very hard. If you have good recommenataions for some fast segmentation models, maybe I could write a demo in the future.

from unitrack.

Thanks for your reply, it helps me lot!

from unitrack.