9ebc0abc8e85a2fa722aaaf75f8c8793e95ea12f,tools/sort.py,Sort,sort_face,#Sort#,167

Before Change

        logger.info("Sorting by identity similarity...")
        filename_list, image_list = self._get_images()

        logger.info("Calculating face identifiers...")
        preds = np.array([self.vgg_face.predict(img)
                          for img in tqdm(image_list, desc="Calculating...", file=sys.stdout)])

        logger.info("Sorting by ward linkage...")

After Change

        logger.info("Sorting by identity similarity...")

        // TODO This should be set in init
        self._loader = ImagesLoader(self.args.input_dir)

        filenames = []
        preds = np.empty((self._loader.count, 512), dtype="float32")
        for idx, (filename, image) in enumerate(tqdm(self._loader.load(),
                                                     desc="Classifying Faces...",
                                                     total=self._loader.count)):
            filenames.append(filename)
            preds[idx] = self.vgg_face.predict(image)

        logger.info("Sorting by ward linkage...")

        indices = self.vgg_face.sorted_similarity(preds, method="ward")
        img_list = np.array(filenames)[indices]

Instances