imatge-upc / rvos

RVOS: End-to-End Recurrent Network for Video Object Segmentation (CVPR 2019)
https://imatge-upc.github.io/rvos/
Other
277 stars 54 forks source link

Initial Colab Notebook Demonstrating Inference #38

Open AlexanderMath opened 3 years ago

AlexanderMath commented 3 years ago

Great work. It takes some time (1-2 hours) to get all dependencies right and use your model for inference. I made a notebook so anyone can use it for inference in ~ 5 min. Let me know if I broke anything, to make it easier the notebook uses zeroshot.

https://colab.research.google.com/drive/1gr7RyiIZtFXgi4789iO1u1jvkEleAOA7?usp=sharing

image

AlexanderMath commented 3 years ago

In the demo it takes 1 iteration a second and has 2-4% GPU utilization.

image

mariocorradetti commented 2 years ago

Thank you, this is very useful. You have also a full implementation of the model, becuase i wanna train this network on a custom dataset using Colab. Thank you in advance