(self,
ginfo,
ann_file,
img_prefix,
data_cfg,
pipeline=None,
test_mode=False,
**kwargs)
| 28 | print("ceph can not be used") |
| 29 | class MSCoCoDataset(Dataset): |
| 30 | def __init__(self, |
| 31 | ginfo, |
| 32 | ann_file, |
| 33 | img_prefix, |
| 34 | data_cfg, |
| 35 | pipeline=None, |
| 36 | test_mode=False, |
| 37 | **kwargs): |
| 38 | self.image_info = {} |
| 39 | self.ann_info = {} |
| 40 | self.initialized = False |
| 41 | |
| 42 | self.use_ceph = True |
| 43 | self.annotations_path = ann_file |
| 44 | self.img_prefix = img_prefix |
| 45 | self.test_mode = test_mode |
| 46 | print('data_cfg0',data_cfg) |
| 47 | # data_cfg=demjson.decode(data_cfg) |
| 48 | # print('data_cfg',data_cfg) |
| 49 | self.ann_info['image_size'] = np.array(data_cfg['image_size']) |
| 50 | self.ann_info['heatmap_size'] = np.array(data_cfg['heatmap_size']) |
| 51 | self.ann_info['num_joints'] = data_cfg['num_joints'] |
| 52 | |
| 53 | self.ann_info['flip_pairs'] = None |
| 54 | |
| 55 | self.ann_info['inference_channel'] = data_cfg['inference_channel'] |
| 56 | self.ann_info['num_output_channels'] = data_cfg['num_output_channels'] |
| 57 | self.ann_info['dataset_channel'] = data_cfg['dataset_channel'] |
| 58 | |
| 59 | self.db = [] |
| 60 | self.task_name = ginfo.task_name |
| 61 | |
| 62 | |
| 63 | pipeline = [ |
| 64 | LoadImageFromFile(), |
| 65 | TopDownRandomFlip(flip_prob=0.5), |
| 66 | TopDownHalfBodyTransform(num_joints_half_body=8,prob_half_body=0.3), |
| 67 | TopDownGetRandomScaleRotation(rot_factor=40, scale_factor=0.5), |
| 68 | TopDownAffine(), |
| 69 | ToTensor(), |
| 70 | NormalizeTensor(mean=[0.485, 0.456, 0.406],std=[0.229, 0.224, 0.225]), |
| 71 | TopDownGenerateTarget(sigma=2), |
| 72 | ] |
| 73 | self.pipeline = ComposeX(pipeline) |
| 74 | |
| 75 | |
| 76 | |
| 77 | |
| 78 | self.use_gt_bbox = data_cfg['use_gt_bbox'] |
| 79 | self.bbox_file = data_cfg['bbox_file'] |
| 80 | self.det_bbox_thr = data_cfg.get('det_bbox_thr', 0.0) |
| 81 | if 'image_thr' in data_cfg: |
| 82 | warnings.warn( |
| 83 | 'image_thr is deprecated, ' |
| 84 | 'please use det_bbox_thr instead', DeprecationWarning) |
| 85 | self.det_bbox_thr = data_cfg['image_thr'] |
| 86 | self.use_nms = data_cfg.get('use_nms', True) |
| 87 | self.soft_nms = data_cfg['soft_nms'] |
nothing calls this directly
no test coverage detected