8d4bae398068c94d616b244f6354f70f7d12e42d,chainercv/datasets/cub/cub_keypoint_dataset.py,CUBKeypointDataset,get_example,#CUBKeypointDataset#Any#,93

Before Change

            else:
                return img, keypoint, kp_mask

        path, _ = os.path.splitext(self.paths[i])
        mask = utils.read_image(
            os.path.join(self.mask_dir, path + ".png"),
            dtype=np.uint8,

After Change

        prob_map = utils.read_image(self.prob_map_paths[i],
                                    dtype=np.uint8, color=False)
        prob_map = prob_map.astype(np.float32) / 255  // [0, 255] -> [0, 1]
        prob_map = prob_map[0]  // (1, H, W) --> (H, W)
        if self.return_bb:
            return img, keypoint, kp_mask, self.bbs[i], prob_map
        else:

Instances