| 914 | return input_query_label, input_query_bbox, attn_mask, attn_mask2, attn_mask3, mask_dict |
| 915 | |
| 916 | def dn_post_process2(self, outputs_class, outputs_coord, |
| 917 | outputs_body_keypoints_list, mask_dict): |
| 918 | if mask_dict and mask_dict['pad_size'] > 0: |
| 919 | output_known_class = [ |
| 920 | outputs_class_i[:, :mask_dict['pad_size'], :] |
| 921 | for outputs_class_i in outputs_class |
| 922 | ] |
| 923 | output_known_coord = [ |
| 924 | outputs_coord_i[:, :mask_dict['pad_size'], :] |
| 925 | for outputs_coord_i in outputs_coord |
| 926 | ] |
| 927 | |
| 928 | outputs_class = [ |
| 929 | outputs_class_i[:, mask_dict['pad_size']:, :] |
| 930 | for outputs_class_i in outputs_class |
| 931 | ] |
| 932 | outputs_coord = [ |
| 933 | outputs_coord_i[:, mask_dict['pad_size']:, :] |
| 934 | for outputs_coord_i in outputs_coord |
| 935 | ] |
| 936 | outputs_keypoint = outputs_body_keypoints_list |
| 937 | |
| 938 | mask_dict.update({ |
| 939 | 'output_known_coord': output_known_coord, |
| 940 | 'output_known_class': output_known_class |
| 941 | }) |
| 942 | return outputs_class, outputs_coord, outputs_keypoint |
| 943 | |
| 944 | def forward(self, data_batch: NestedTensor, targets: List = None): |
| 945 | """The forward expects a NestedTensor, which consists of: |