3b2437d297a567ba16877ab7b6a9b0ff4ddd4914,chainercv/datasets/cub/cub_keypoint_dataset.py,CUBKeypointDataset,get_example,#CUBKeypointDataset#Any#,84

Before Change

        keypoint = np.array(self.kp_dict[i], dtype=np.float32)
        kp_mask = np.array(self.kp_mask_dict[i], dtype=np.bool)

        if self.crop_bbox:
            // (y_min, x_min, y_max, x_max)
            bbox = self.bboxes[i].astype(np.int32)
            img = img[:, bbox[0]: bbox[2], bbox[1]: bbox[3]]
            keypoint[:, :2] = keypoint[:, :2] - bbox[:2]

        if not self.return_mask:
            return img, keypoint, kp_mask

        path, _ = os.path.splitext(self.paths[i])

After Change

            os.path.join(self.mask_dir, path + ".png"),
            dtype=np.uint8,
            color=False)
        if self.return_bb:
            return img, keypoint, kp_mask, self.bbs[i], mask
        else:
            return img, keypoint, kp_mask, mask

Instances