def __call__(self, **data_dict):
     data_dict = convert_seg_to_bounding_box_coordinates(
         data_dict,
         self.dim,
         self.get_rois_from_seg_flag,
         class_specific_seg_flag=self.class_specific_seg_flag)
     return data_dict
 def __call__(self, **data_dict):
     data_dict['bb_target'], data_dict['roi_masks'], data_dict['roi_labels'] = convert_seg_to_bounding_box_coordinates(
         data_dict, self.dim, self.get_rois_from_seg)
     fg_bg_seg = np.copy(data_dict['seg']) # roi encoded information not needed anymore. map to foreground background array.
     fg_bg_seg[fg_bg_seg > 0 ] = 1
     data_dict['seg'] = fg_bg_seg
     return data_dict
Exemplo n.º 3
0
 def __call__(self, **data_dict):
     data_dict['bb_target'], data_dict['roi_masks'], data_dict[
         'roi_labels'] = convert_seg_to_bounding_box_coordinates(
             data_dict, self.dim, self.get_rois_from_seg)
     fg_bg_seg = np.copy(
         data_dict['seg']
     )  # roi encoded information not needed anymore. map to foreground background array.
     fg_bg_seg[fg_bg_seg > 0] = 1
     data_dict['seg'] = fg_bg_seg
     return data_dict
Exemplo n.º 4
0
 def __call__(self, **data_dict):
     data_dict['bb_target'], data_dict['roi_masks'], data_dict[
         'roi_class_ids'] = convert_seg_to_bounding_box_coordinates(
             data_dict['seg'], data_dict['class_target'], data_dict['pid'],
             self.dim)
     data_dict['patient_target'] = data_dict[
         'class_target'] + 1  # add background class
     data_dict['class_target'] = [
         data_dict['roi_class_ids'], data_dict['roi_masks']
     ]
     return data_dict