hkchengrex / XMem

[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model

Home Page:https://hkchengrex.com/XMem/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Make a custom dataset

wang-shuaikang opened this issue · comments

Hello,
Thank you for your great work. I want to make a dataset by myself. Do I need to provide a video and the mask of the first frame of the video, and then according to which code in the project can I generate my object mask dataset? ? Can it be generated through the interactive_demo.py file? I would be very grateful if you could answer! ! !

Using eval.py with https://github.com/hkchengrex/XMem/blob/main/docs/INFERENCE.md#on-custom-data should be the easiest way. You would need the mask of the first frame.