def test_image_across_descriptor(image_fname): img = cv2.imread(image_fname) first_description = None for i, key in enumerate(dd.keys()): methodToCall = getattr(descriptor_definitions, key) descriptor = methodToCall() preprocessed = descriptor.do_preprocess(img) description = descriptor.describe(img) if first_description is None: first_description = description distance = EuclideanComparator().compare(first_description["value"], description["value"]) print "distance=%s" % distance plt.subplot(2, 3, i + 1), plt.imshow(preprocessed, 'gray') plt.title("%s_%s" % (key[0], image_fname))
def test_datasets(): total = -1 for key in dd.keys(): total = test_dataset(key, total + 1)
# i = descriptor.do_preprocess(img) # cv2.imshow("test", i) print "fname=%s" % fname description = descriptor.describe(img) if description is not None: description["value"] = description["value"].tolist() data[fname[:-4]] = description descriptor_data["data"] = data descriptor_fname = "datasets/%s" % (key) with open(descriptor_fname, 'w') as outfile: json.dump(descriptor_data, outfile, indent=2) for key in dd.keys(): descriptor_fname = "datasets/%s" % (key) if not os.path.isfile(descriptor_fname): create_ds(key) # def create_dataset(create_descriptor, dname): # dataset_fname = "datasets/%s" % dname # if not os.path.isfile(dataset_fname): # data = {} # for fname in os.listdir("data/"): # print "fname=%s" % fname # if ".jpg" in fname: # image_detail ={ # "fname": "data/%s" % fname # } # # img = cv2.imread("data/%s" % fname)
# cv2.imshow("test", i) print "fname=%s" % fname description = descriptor.describe(img) if description is not None: description["value"] = description["value"].tolist() data[fname[:-4]] = description descriptor_data["data"] = data descriptor_fname = "datasets/%s" % (key) with open(descriptor_fname, 'w') as outfile: json.dump(descriptor_data, outfile, indent=2) for key in dd.keys(): descriptor_fname = "datasets/%s" % (key) if not os.path.isfile(descriptor_fname): create_ds(key) # def create_dataset(create_descriptor, dname): # dataset_fname = "datasets/%s" % dname # if not os.path.isfile(dataset_fname): # data = {} # for fname in os.listdir("data/"): # print "fname=%s" % fname # if ".jpg" in fname: # image_detail ={ # "fname": "data/%s" % fname # }