search

hkchengrex / Mask-Propagation

[CVPR 2021] MiVOS - Mask Propagation module. Reproduced STM (and better) with training code :star2:. Semi-supervised video object segmentation evaluation.
https://hkchengrex.github.io/MiVOS/
MIT License
127 stars 22 forks source link

about train stage #41

Closed lsy-dot closed 2 years ago

lsy-dot commented 2 years ago

Sorry to bother you, I plan to use my own image dataset and video dataset for training, I would like to ask if I can skip the training of stage1, i.e. use the image dataset to train stage0, use the video dataset (I have converted into the DAVIS dataset format) training stage2

hkchengrex commented 2 years ago

So are you skipping stage 0 or stage 1? Either is fine though. I would recommend STCN if you are working on something new as it is better than the model in this repo in almost every way. The code structure is similar so it should be easy to follow from here.

lsy-dot commented 2 years ago

Thank you for your prompt reply! I want to skip stage1 training, and thanks for the advice and great work!