hkchengrex / Cutie

[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation
https://hkchengrex.com/Cutie/
MIT License
695 stars 70 forks source link

Cutie output masks convert to OVIS dataset format #110

Closed lpy523 closed 1 month ago

lpy523 commented 1 month ago

Thanks for your great contribution! I was wondering if there is a way to convert Cutie's output mask to a format similar to the OVIS dataset for training video instance segmentation models? Thanks!

hkchengrex commented 1 month ago

Yes. IIRC it uses the COCO format so you can use pycocotools: https://github.com/ppwwyyxx/cocoapi