demo inference speed is very slow

berkeley-hipie / HIPIE

[NeurIPS2023] Code release for "Hierarchical Open-vocabulary Universal Image Segmentation"

MIT License

260 stars 19 forks source link

Open dreamer121121 opened 1 year ago

dreamer121121 commented 1 year ago

excellent work!, i tried the hipie-main demo, but i found the speed is very slow about 1min for a img, could you give some advices?

jacklishufan commented 1 year ago

the default demo notebook is using cpu. Try use GPU instead by changing device='cpu' to device='cuda'