def sort_hist(self):
Sort by image histogram similarity
logger.info("Sorting by histogram similarity...")
filename_list, image_list = self._get_images()
distance = cv2.HISTCMP_BHATTACHARYYA
logger.info("Calculating histograms...")
histograms = [cv2.calcHist([img], [0], None, [256], [0, 256]) for img in image_list]
img_list = list(zip(filename_list, histograms))
logger.info("Comparing histograms and sorting...")
After Change
logger.info("Sorting by histogram similarity...")
// TODO We have metadata here, so we can mask the face for hist sorting
img_list = [(filename, cv2.calcHist([image], [0], None, [256], [0, 256]))
for filename, image, _ in tqdm(self._loader.load(),
desc="Calculating histograms",
total=self._loader.count,
leave=False)]