keypoint_data.py 文件源码

def load(file_name):
        data = pickle.load(open(file_name, "rb" ))

        keypoints = []
        descriptors = []

        for entry in data:
            point = entry[0]
            size = entry[1]
            angle = entry[2]
            response = entry[3]
            octave = entry[4]
            class_id = entry[5]

            keypoints.append(cv2.KeyPoint(x=point[0],y=point[1]
                , _size=size
                , _angle=angle
                , _response=response
                , _octave=octave
                , _class_id=class_id))

            descriptors.append(entry[6])

        return KeypointData(keypoints, np.array(descriptors, np.float32))