| 2196 | |
| 2197 | |
| 2198 | class AiOSSMPLX_Box(nn.Module): |
| 2199 | def __init__( |
| 2200 | self, |
| 2201 | backbone, |
| 2202 | transformer, |
| 2203 | num_classes, |
| 2204 | num_queries, |
| 2205 | aux_loss=False, |
| 2206 | iter_update=True, |
| 2207 | query_dim=4, |
| 2208 | random_refpoints_xy=False, |
| 2209 | fix_refpoints_hw=-1, |
| 2210 | num_feature_levels=1, |
| 2211 | nheads=8, |
| 2212 | two_stage_type='no', |
| 2213 | dec_pred_class_embed_share=False, |
| 2214 | dec_pred_bbox_embed_share=False, |
| 2215 | dec_pred_pose_embed_share=False, |
| 2216 | two_stage_class_embed_share=True, |
| 2217 | two_stage_bbox_embed_share=True, |
| 2218 | dn_number=100, |
| 2219 | dn_box_noise_scale=0.4, |
| 2220 | dn_label_noise_ratio=0.5, |
| 2221 | dn_batch_gt_fuse=False, |
| 2222 | dn_labelbook_size=100, |
| 2223 | dn_attn_mask_type_list=['group2group'], |
| 2224 | cls_no_bias=False, |
| 2225 | num_group=100, |
| 2226 | num_body_points=0, |
| 2227 | num_hand_points=0, |
| 2228 | num_face_points=0, |
| 2229 | num_box_decoder_layers=2, |
| 2230 | num_hand_face_decoder_layers=4, |
| 2231 | body_model=dict( |
| 2232 | type='smplx', |
| 2233 | keypoint_src='smplx', |
| 2234 | num_expression_coeffs=10, |
| 2235 | keypoint_dst='smplx_137', |
| 2236 | model_path='data/body_models/smplx', |
| 2237 | use_pca=False, |
| 2238 | use_face_contour=True), |
| 2239 | train=True, |
| 2240 | inference=False, |
| 2241 | focal_length=[5000., 5000.], |
| 2242 | camera_3d_size=2.5 |
| 2243 | ): |
| 2244 | super().__init__() |
| 2245 | self.num_queries = num_queries |
| 2246 | self.transformer = transformer |
| 2247 | self.num_classes = num_classes |
| 2248 | self.hidden_dim = hidden_dim = transformer.d_model |
| 2249 | self.num_feature_levels = num_feature_levels |
| 2250 | self.nheads = nheads |
| 2251 | self.label_enc = nn.Embedding(dn_labelbook_size + 1, hidden_dim) |
| 2252 | self.num_body_points = num_body_points |
| 2253 | self.num_hand_points = num_hand_points |
| 2254 | self.num_face_points = num_face_points |
| 2255 | self.num_whole_body_points = num_body_points + 2*num_hand_points + num_face_points |
no outgoing calls
no test coverage detected