def _generate_class_vector(self, category, bow_files, positives): positives = [util.bow_name(p) for p in positives] # log("Number of positive examples: ", len(positives)) y = np.zeros(len(bow_files), dtype=np.int32) indices = np.array([f in positives for f in bow_files], dtype=np.bool) y[indices] = 1 return y
def test_bow_descriptor_name(self): self.assertEqual(util.bow_name("/test/test2/{1234}.jpg"), "/test/test2/{1234}.descr_bowdescr.bin") self.assertEqual(util.bow_name("{1234}.jpg"), "{1234}.descr_bowdescr.bin") self.assertEqual(util.bow_name("{1234}"), "{1234}.descr_bowdescr.bin")
def get_bow_filenames(self, file_names): files = set(os.listdir(self.PATHS["BOW"])) bows = [util.bow_name(f) for f in file_names] return [b for b in bows if b in files]