ec23dc7d3e64f5413ef061fb8f4fe1ae556bb434,chainercv/datasets/cub/cub_keypoint_dataset.py,CUBKeypointDataset,__init__,#CUBKeypointDataset#Any#Any#Any#Any#,67

Before Change

    def __init__(self, data_dir="auto", return_bb=False,
                 prob_map_dir="auto", return_prob_map=False):
        super(CUBKeypointDataset, self).__init__(data_dir, prob_map_dir)
        self.data_names = ("img", "keypoint", "kp_mask")
        if return_bb:
            self.data_names += ("bb",)
        if return_prob_map:
            self.data_names += ("prob_map",)
        self.add_getter(("keypoint", "kp_mask"), self.get_kp)

        // load keypoint
        parts_loc_file = os.path.join(self.data_dir, "parts", "part_locs.txt")

After Change

        self.add_getter(("keypoint", "kp_mask"), self.get_kp)

        keys = ["img", "keypoint", "kp_mask"]
        if return_bb:
            keys.append("bb")
        if return_prob_map:
            keys.append("prob_map")
        self.keys = keys

    def get_kp(self, i):
        Returns the keypoints and their mask of the i-th example.

Instances