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]
