def doScan(self): if self._windowIsClosing: return self.testImg = None self.updateDisplay_() rec = FaceRec(self.training_images_path, self.src_dir) filtered_images = rec.filter() self.queue = [{'img_path': img_path, 'label': label, 'confidence': confidence} for img_path,label,confidence in filtered_images] self.performSelectorOnMainThread_withObject_waitUntilDone_("updateDisplay:", 1, 0)
from face_rec import FaceRec import os import sys if __name__ == "__main__": if len(sys.argv) != 3: print("Usage: %s train_images/ src_path_to_filter/" % (sys.argv[0])) exit(1) training_images_path = sys.argv[1] src_images_path = sys.argv[2] try: print("Starting....") rec = FaceRec(training_images_path, src_images_path) filtered_images = rec.filter() print(filtered_images) except Exception as error: print(error) exit(0)