GPU support

[caddyless]

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]

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]

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.