opencv_transforms_torchvision icon indicating copy to clipboard operation
opencv_transforms_torchvision copied to clipboard

Published 20 hours ago • verified publisher icon hityzy1122

GPU support

Open caddyless opened this issue 4 years ago • 2 comments

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.

caddyless avatar Oct 22 '20 07:10 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.

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.

hityzy1122 avatar Oct 22 '20 07:10 hityzy1122

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.

caddyless avatar Oct 22 '20 07:10 caddyless