seg_dataset.py 文件源码

def get_full_size_labels(self, img_ids, timespan=None):
    """Get full sized labels."""
    if timespan is None:
      timespan = self.get_default_timespan()
    with h5py.File(self.h5_fname, "r") as h5f:
      num_ex = len(img_ids)
      y_full = []
      for kk, ii in enumerate(img_ids):
        key = self.get_str_id(ii)
        data_group = h5f[key]
        if "label_ins_seg_full" in data_group:
          y_gt_group = data_group["label_ins_seg_full"]
          num_obj = len(y_gt_group.keys())
          y_full_kk = None
          for jj in range(min(num_obj, timespan)):
            y_full_jj_str = y_gt_group["{:03d}".format(jj)][:]
            y_full_jj = cv2.imdecode(
                y_full_jj_str, cv2.CV_LOAD_IMAGE_GRAYSCALE).astype(np.float32)
            if y_full_kk is None:
              y_full_kk = np.zeros(
                  [timespan, y_full_jj.shape[0], y_full_jj.shape[1]])
            y_full_kk[jj] = y_full_jj
          y_full.append(y_full_kk)
        else:
          y_full.append(np.zeros([timespan] + list(data_group["orig_size"][:])))
    return y_full