Visibility

Hello,

I am considering to use this model to label my unlabeled dataset. I am wondering how can I get the visibilty. The 'pred_keypoints' has shape n_query, 204. I am wondering if the 34th to 51st in the second dimension corresponds to the visibility.