search

hityzy1122 / opencv_transforms_torchvision

opencv reimplement for transforms in torchvision
MIT License
193 stars 29 forks source link

GPU support #15

Open caddyless opened 3 years ago

caddyless commented 3 years ago

Thanks for this greate package! Is there a plan for adding GPU support for this package? I means, opencv has already added gpu support to many image operations.

hityzy1122 commented 3 years ago

Thanks for this greate package! Is there a plan for adding GPU support for this package? I means, opencv has already added gpu support to many image operations.

That's a good suggestion! however, it's a pitty that, I am working as an intern in senseitme and do not have much time to build a GPU version. There is a GPU based lib https://github.com/NVIDIA/DALI which you may be interested in.

caddyless commented 3 years ago

Thanks a lot whatever. This package is quite friendly to pytorchers, since we just need make little adjustment on our existed projects. DALI is a famous GPU based image process lib. However, it also has much more steep learning curve. Besides that, it is said that there is a bug on memory usage of DALI (the usage of memory will increase as the iteration grow.) which will limit the maximum batch size. This claim also escapes me from it. That's why I appreciate this package so much.