BagOfWords.py 文件源码

def createTestingInstances(self, images):

        cv2.ocl.setUseOpenCL(False)
        orb = cv2.ORB_create()
        instances = []

        for img, label in images:
            print img
            img = read_color_image(img)

            keypoints = orb.detect(img, None)
            keypoints, descriptors = orb.compute(img, keypoints)

            if descriptors is None:
                descriptors = []

            histogram = np.zeros(self.center_num)
            for d in descriptors:
                min_val = None
                min_index = None
                for j in xrange(len(self.centers)):
                    distance = np.linalg.norm(d - self.centers[j])
                    if min_val is None or distance < min_val:
                        min_val = distance
                        min_index = j
                histogram.itemset(min_index, 1 + histogram.item(min_index))
            instances.append(Instance(histogram, label))

        self.testing_instances = instances