hkchengrex / Tracking-Anything-with-DEVA

[ICCV 2023] Tracking Anything with Decoupled Video Segmentation
https://hkchengrex.com/Tracking-Anything-with-DEVA/
Other
1.23k stars 128 forks source link

extracting mask and bbox #83

Closed shani1610 closed 5 months ago

shani1610 commented 5 months ago

Hi, Thank you for this work. I'm using the notebook you've shared and I'm trying to extract and save the masks and bounding boxes values, how can i do it? straight from result_saver?

here is the snippet from the collab: image

Thank you!

hkchengrex commented 5 months ago

Yes. You would need to add a new mode in result_saver and specify that as dataset during ResultSaver initialization.