SHI-Labs / OneFormer

OneFormer: One Transformer to Rule Universal Image Segmentation, arxiv 2022 / CVPR 2023
https://praeclarumjj3.github.io/oneformer
MIT License
1.39k stars 129 forks source link

Update colab demo #108

Open bakachan19 opened 6 months ago

bakachan19 commented 6 months ago

Hi, thanks for this great work. In the colab demo I noticed that a different repo is being cloned: !git clone https://github.com/SHI-Labs/OneFormer-Colab.git

instead of: https://github.com/SHI-Labs/OneFormer.git.

Also, the yaml configurations are set up for OneFormer-Colab.git.

Which models/configurations give the best results: OneFormer-Colab.git or OneFormer.git?

Would it be possible to update the colab demo with configurations for the latest/best models?

Later edit: this requirement -f https://shi-labs.com/natten/wheels/cu113/torch1.10.1/index.html does not work in the current colab version. Could you also update the installation dependencies in colab?

Thanks!

immengzi commented 1 month ago

this requirement -f https://shi-labs.com/natten/wheels/cu113/torch1.10.1/index.html does not work in the current colab version

Well, you may spend over 60 minutes in building natten. I just waited for a long time and then succeeded...