About using GPU to accelerate trapped ball algorithm

Hello, I find it very slow when I use gen_labelmap.py to generate labelmap. I wonder how we can accelerate the procedure with GPU or any clue on it?
Thank you for your Great works