search

hkchengrex / XMem

[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model
https://hkchengrex.com/XMem/
MIT License
1.72k stars 191 forks source link

Make a custom dataset #114

Closed wang-shuaikang closed 1 year ago

wang-shuaikang commented 1 year ago

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! ! !

hkchengrex commented 1 year ago

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.