@@ -99,15 +99,15 @@ def __init__(self, transform, data_split):
9999 def load_data (self , train_sample_interval = 1 ):
100100
101101 content = np .load (self .annot_path , allow_pickle = True )
102- # import ipdb;ipdb.set_trace()
102+
103103 try :
104104 frame_range = content ['frame_range' ]
105105 except KeyError :
106106 frame_range = \
107107 np .array ([[i , i + 1 ] for i in range (self .num_data )])
108108
109109 num_examples = len (frame_range )
110- # import ipdb;ipdb.set_trace()
110+
111111 if 'meta' in content :
112112 meta = content ['meta' ].item ()
113113 print ('meta keys:' , meta .keys ())
@@ -180,7 +180,7 @@ def load_data(self, train_sample_interval=1):
180180 valid_kps3d = False
181181 keypoints3d_mask = None
182182 valid_kps3d_mask = False
183- # import ipdb;ipdb.set_trace()
183+
184184
185185 # processing keypoints
186186 for kps3d_key in KPS3D_KEYS :
@@ -205,7 +205,7 @@ def load_data(self, train_sample_interval=1):
205205 if keypoints2d .shape [- 1 ] == 3 :
206206 valid_kps3d_mask = True
207207 occlusion = content ['meta' ][()]['occ' ] if 'occ' in content ['meta' ][()] and len (content ['meta' ][()]['occ' ])> 0 else None
208- # import ipdb;ipdb.set_trace()
208+
209209 print ('Done. Time: {:.2f}s' .format (time .time () - tic ))
210210
211211 datalist = []
@@ -223,17 +223,17 @@ def load_data(self, train_sample_interval=1):
223223
224224
225225 bbox_list = bbox_xywh [frame_start :frame_end , :4 ]
226- # import ipdb;ipdb.set_trace()
226+
227227 valid_idx = []
228228 body_bbox_list = []
229229
230230 if hasattr (cfg , 'bbox_ratio' ):
231231 bbox_ratio = cfg .bbox_ratio * 0.833 # preprocess body bbox is giving 1.2 box padding
232232 else :
233233 bbox_ratio = 1.25
234- # import ipdb;ipdb.set_trace()
234+
235235 for bbox_i , bbox in enumerate (bbox_list ):
236- # import ipdb;ipdb.set_trace()
236+
237237 bbox = process_bbox (bbox ,
238238 img_width = img_shape [1 ],
239239 img_height = img_shape [0 ],
@@ -293,7 +293,7 @@ def load_data(self, train_sample_interval=1):
293293 lhand_bbox_list .append (lhand_bbox )
294294 rhand_bbox_list .append (rhand_bbox )
295295 face_bbox_list .append (face_bbox )
296- # import ipdb;ipdb.set_trace()
296+
297297 # lhand_bbox = np.stack(lhand_bbox_list,axis=0)
298298 # rhand_bbox = np.stack(rhand_bbox_list,axis=0)
299299 # face_bbox = np.stack(face_bbox_list,axis=0)
@@ -305,7 +305,7 @@ def load_data(self, train_sample_interval=1):
305305 joint_cam = keypoints3d [valid_idx ]
306306 else :
307307 joint_cam = None
308- # import ipdb;ipdb.set_trace()
308+
309309 if 'leye_pose_0' in smplx .keys ():
310310 smplx .pop ('leye_pose_0' )
311311 if 'leye_pose_1' in smplx .keys ():
@@ -517,10 +517,10 @@ def __getitem__(self, idx):
517517 # for name in ('L_Big_toe', 'L_Small_toe', 'L_Heel', 'R_Big_toe', 'R_Small_toe', 'R_Heel'):
518518 # smplx_joint_valid[smpl_x.joints_name.index(name)] = 0
519519 smplx_joint_valid = smplx_joint_valid [:, :, None ]
520- # import ipdb;ipdb.set_trace()
520+
521521
522522
523- # import ipdb;ipdb.set_trace()
523+
524524 # TODO: check here
525525 # if not (smplx_shape == 0).all():
526526 # smplx_shape_valid = True
@@ -543,7 +543,7 @@ def __getitem__(self, idx):
543543 body_bbox_valid_list = []
544544 body_bbox_list = []
545545 # hand and face bbox transform
546- # import ipdb;ipdb.set_trace()
546+
547547
548548 for i in range (num_person ):
549549 lhand_bbox , lhand_bbox_valid = self .process_hand_face_bbox (
@@ -555,7 +555,7 @@ def __getitem__(self, idx):
555555 face_bbox , face_bbox_valid = self .process_hand_face_bbox (
556556 data ['face_bbox' ][i ], do_flip , img_shape , img2bb_trans ,
557557 cropped_img_shape )
558- # import ipdb;ipdb.set_trace()
558+
559559 body_bbox , body_bbox_valid = self .process_hand_face_bbox (
560560 data ['bbox' ][i ], do_flip , img_shape , img2bb_trans ,
561561 cropped_img_shape )
@@ -590,7 +590,7 @@ def __getitem__(self, idx):
590590 body_bbox_center_list .append (body_bbox_center )
591591 body_bbox_size_list .append (body_bbox_size )
592592 body_bbox_valid_list .append (body_bbox_valid )
593- # import ipdb;ipdb.set_trace()
593+
594594
595595 body_bbox = np .stack (body_bbox_list , axis = 0 )
596596 lhand_bbox = np .stack (lhand_bbox_list , axis = 0 )
@@ -613,7 +613,7 @@ def __getitem__(self, idx):
613613 if 'occlusion' in data :
614614 occlusion = data ['occlusion' ]
615615 occ_mask = occlusion < 97
616- # import ipdb;ipdb.set_trace()
616+
617617 joint_img_aug [:,:,2 ] = joint_img_aug [:,:,2 ]* occ_mask [:,None ]
618618 joint_cam_wo_ra [:,:,3 ] = joint_cam_wo_ra [:,:,3 ]* occ_mask [:,None ]
619619 joint_trunc = joint_trunc * occ_mask [:,None ,None ]
@@ -641,8 +641,8 @@ def __getitem__(self, idx):
641641 if self .__class__ .__name__ == 'GTA_Human2' :
642642 smplx_shape_valid = smplx_shape_valid * 0
643643 # if body_bbox_valid.sum() > 0:
644- # import ipdb;ipdb.set_trace()
645- # import ipdb;ipdb.set_trace()
644+
645+
646646 targets = {
647647 # keypoints2d, [0,img_w],[0,img_h] -> [0,1] -> [0,output_hm_shape]
648648 'joint_img' : joint_img_aug [body_bbox_valid > 0 ],
@@ -761,7 +761,7 @@ def __getitem__(self, idx):
761761 if self .data_split == 'test' :
762762 self .cam_param = {}
763763 joint_cam = data ['joint_cam' ]
764- # import ipdb;ipdb.set_trace()
764+
765765 if joint_cam is not None :
766766 dummy_cord = False
767767 joint_cam [:,:,:3 ] = joint_cam [:,:,:3 ] - joint_cam [
@@ -836,7 +836,7 @@ def __getitem__(self, idx):
836836 face_bbox , face_bbox_valid = self .process_hand_face_bbox (
837837 data ['face_bbox' ][i ], do_flip , img_shape , img2bb_trans ,
838838 cropped_img_shape )
839- # import ipdb;ipdb.set_trace()
839+
840840 body_bbox , body_bbox_valid = self .process_hand_face_bbox (
841841 data ['bbox' ][i ], do_flip , img_shape , img2bb_trans ,
842842 cropped_img_shape )
0 commit comments