search

haruishi43 / equilib

🌎→🗾Equirectangular (360/panoramic) image processing library for Python with minimal dependencies only using Numpy and PyTorch
Apache License 2.0
163 stars 21 forks source link

enabling gpu in equi2per transform #9

Closed zhuqiangLu closed 2 years ago

zhuqiangLu commented 2 years ago

Hi

THANK YOU for your work and it just saved my life.

I am wondering how to turn on the gpu support when I transform ERP to perspective image

haruishi43 commented 2 years ago

Hi @zhuqiangLu , I'm glad my work helped you.

If the inputs are torch.tensor and they're already on the GPU, it will automatically run in GPU mode.

zhuqiangLu commented 2 years ago

Nice to hear this, thank you so much