Hi, the authors of the original paper had masking + key point detection. Is this possible with your code?