コード例 #1
0
 def __init__(self):
     self.config = ConfigReader()
     self.logger = LoggerFactory()
     self.haarDetector = HaarFaceDetector()
     self.dnnDetector = DnnFaceDetector()
     self.azureDetectionClient = AzureFaceClient()
     self.detectionTypes = DetectionTypes()
コード例 #2
0
class CameraOperator():
    def __init__(self):
        self.logger = LoggerFactory()

    def init_camera(self):
        if PI_CAMERA_USED:
            print("Picamera cannot be launch on windows")
            raise Exception("Picamera cannot be launch on windows")
        # COnfiguration for Raspberry
        # camera = PiCamera()
        # camera.resolution = tuple(conf["resolution"])
        # camera.framerate = conf["fps"]
        # rawCapture = PiRGBArray(camera, size=tuple(conf["resolution"]))
        else:
            self.camera = cv2.VideoCapture(CAMERA_PORT)
        time.sleep(CAMERA_WARMUP_TIME)

    def release_camera(self):
        self.camera.release()
        cv2.destroyAllWindows()

    def grab_frame(self):
        grabbed, frame = self.camera.read()
        if not grabbed:
            self.logger.error("Camera didn't grab a frame, raising error")
            raise Exception("Camera didn't grab a frame, raising error")
        return frame

    def show_video(self, frame):
        cv2.imshow("Security Feed", frame)
コード例 #3
0
 def __init__(self):
     self.logger = LoggerFactory()
     self.config = ConfigReader()
     self.request_manager = DetectionRequestsManager()
     self.directory = DirectoryManager()
     self.faceDetectionRepository = FaceDetectionRepository()
     self.pathsProvider = PathsProvider()
コード例 #4
0
 def __init__(self):
     self.logger = LoggerFactory()
     self.pathsProvider = PathsProvider()
     self.filesUploader = FilesUploader()
     self.nnFilesRepo = NeuralNetworkFileRepository()
     self.nnTypes = NeuralNetworkTypes()
     self.stringOperator = StringOperator()
コード例 #5
0
class AzureLargeGroupTrainer():
    def __init__(self):
        self.logger = LoggerFactory()
        self.largeGroupClient = AzureLargeGroupsClient()
        self.nnFilesRepo = NeuralNetworkFileRepository()
        self.nnTypes = NeuralNetworkTypes()

    def train_large_group(self, request_id, name, people_with_image_paths):
        try:
            start = time.time()
            self.largeGroupClient.create_large_group(request_id, name)
            people_ids = set(
                person_id for person_id, image_path in people_with_image_paths)
            people_dictionary = {}
            for person_id in people_ids:
                person_in_group_id = self.largeGroupClient.create_person_in_large_group(
                    request_id, person_id)
                people_dictionary[f'{person_id}'] = person_in_group_id
            for person_id, image_path in people_with_image_paths:
                group_person_id = people_dictionary[f'{person_id}']
                self.largeGroupClient.add_face_to_person_in_large_group(
                    group_person_id, request_id, image_path)
            self.largeGroupClient.train_large_group(request_id)
            end = time.time()
            file_size = '0'
            neural_network_file_entity = NeuralNetworkFile(
                str(request_id), request_id, self.nnTypes.azure_large_group_id,
                str(end - start), '1>', file_size, str(people_dictionary))
            # people dictionary may become problematic when group is very big
            self.nnFilesRepo.add_neural_network_file(
                neural_network_file_entity)

        except Exception as ex:
            self.logger.error(
                f"Exception when creating Azure Large Group. Exception: {ex}")
コード例 #6
0
 def __init__(self):
     self.logger = LoggerFactory()
     self.imagesProvider = PeopleImagesProvider()
     self.recognizersProvider = FaceRecognizerProvider()
     self.openCvRecognizer = OpenCvFaceRecognizerForTesting()
     self.azureRecognizer = AzureFaceRecognizer()
     self.faceDetector = FaceDetectorsManager()
     self.neuralNetworkToUse = 44
コード例 #7
0
 def __init__(self):
     self.imageEditor = ImageEditor()
     self.pathsProvider = PathsProvider()
     self.attachmentTypes = ImageAttachmentTypes()
     self.filesUploader = FilesUploader()
     self.movementRepo = MovementRepository()
     self.directoryManager = DirectoryManager()
     self.logger = LoggerFactory()
コード例 #8
0
class FaceDetectorsManager():
    def __init__(self):
        self.config = ConfigReader()
        self.logger = LoggerFactory()
        self.haarDetector = HaarFaceDetector()
        self.dnnDetector = DnnFaceDetector()
        self.azureDetectionClient = AzureFaceClient()
        self.detectionTypes = DetectionTypes()

    def get_faces_on_image_from_file_path(self, file_path):
        image = cv2.imread(file_path)
        time_before = time.time()
        faces_detected_by_haar = self.haarDetector.run_detector(image)
        time_after_haar = time.time()
        faces_detected_by_dnn = self.dnnDetector.run_detector(image)
        time_after_dnn = time.time()
        faces_detected_by_azure = self.azureDetectionClient.get_face_rectangles(file_path)
        time_after_azure = time.time()
        result = [[self.detectionTypes.haar, faces_detected_by_haar, time_after_haar - time_before],
                  [self.detectionTypes.dnn, faces_detected_by_dnn, time_after_dnn - time_after_haar],
                  [self.detectionTypes.azure, faces_detected_by_azure, time_after_azure - time_after_dnn]]
        self.logger.info(f"Faces detected: \n {result}")
        return result

    def get_faces_on_image(self, image):
        faces_detected_by_haar = self.haarDetector.run_detector(image)
        faces_detected_by_dnn = self.dnnDetector.run_detector(image)
        self.logger.info(f"Faces detected by "
                         f"\n   Haar: {faces_detected_by_haar}"
                         f"\n   DNN: {faces_detected_by_dnn}")
        return faces_detected_by_haar, faces_detected_by_dnn

    def get_face_by_haar(self, image):
        faces_detected_by_haar = self.haarDetector.run_detector(image)
        self.logger.info(f"Faces detected by "
                         f"\n   Haar: {faces_detected_by_haar}")
        return faces_detected_by_haar

    def get_face_by_dnn(self, image):
        faces_detected_by_dnn = self.dnnDetector.run_detector(image)
        self.logger.info(f"Faces detected by "
                         f"\n   DNN: {faces_detected_by_dnn}")
        return faces_detected_by_dnn

    def get_face_by_azure(self, file_path):
        faces_detected_by_azure = self.azureDetectionClient.get_face_rectangles(file_path)
        self.logger.info(f"Faces detected by "
                         f"\n   Azure: {faces_detected_by_azure}")
        return faces_detected_by_azure

    def get_face_by_haar_with_load_image(self, file_path):
        image = cv2.imread(file_path)
        return self.get_face_by_haar(image)

    def get_face_by_dnn_with_load_image(self, file_path):
        image = cv2.imread(file_path)
        return self.get_face_by_dnn(image)
コード例 #9
0
 def __init__(self):
     self.imageEditor = ImageEditor()
     self.filesUploader = FilesUploader()
     self.files_repository = ImageAttachmentRepository()
     self.pathsProvider = PathsProvider()
     self.attachmentTypes = ImageAttachmentTypes()
     self.detectionTypes = DetectionTypes()
     self.resultsRepository = DetectionResultRepository()
     self.logger = LoggerFactory()
コード例 #10
0
class NeuralNetworkRequestsManager():
    def __init__(self):
        self.logger = LoggerFactory()
        self.neuralNetworkRepo = NeuralNetworkRepository()
        self.neuralNetworksTrainingManager = NeuralNetworksTrainingManager()

    def process_request(self, request: NeuralNetwork):
        self.logger.info(f"Working on creating neural network for request with {request.id} id")
        self.neuralNetworksTrainingManager.create_all_neural_networks(request.id, request.name,
                                                                      request.maxNumberOfPhotosPerPerson)
        self.neuralNetworkRepo.complete_request(request.id)
コード例 #11
0
class ProgramRunner:
    def __init__(self):
        self.logger = LoggerFactory()
        self.detectionProcessRunner = FaceDetectionProcess()
        self.recognitionProcessRunner = FaceRecognitionProcess()
        self.neuralNetworkTrainingProcessRunner = NeuralNetworkTrainingRunner()

    def run_program(self):
        self.logger.info("START APP")
        self.detectionProcessRunner.run_face_detection()
        self.neuralNetworkTrainingProcessRunner.run_training()
        self.recognitionProcessRunner.run_face_recognition()
        self.logger.info("END APP")
コード例 #12
0
class RecognitionRequestManager():
    def __init__(self):
        self.logger = LoggerFactory()
        self.inputFileProvider = InputFileProvider()
        self.recognitionRepository = FaceRecognitionRepository()
        self.recognizersManager = FaceRecognizersManager()

    def process_request(self, request: Recognition):
        self.logger.info(f"Working on face recognition request {request.id} id")
        input_file_path = self.inputFileProvider.get_recognition_input_file_path(request.id)
        self.recognizersManager.get_identity_by_open_cv_recognizers(request.id, request.neural_network_id, input_file_path)
        self.recognizersManager.get_identity_by_azure_cognitive(request.id, request.neural_network_id, input_file_path)
        self.logger.info(f"Completed recognition request id: {request.id} name: {request.name}")
コード例 #13
0
class ResultsOperator:
    def __init__(self):
        self.imageEditor = ImageEditor()
        self.filesUploader = FilesUploader()
        self.files_repository = ImageAttachmentRepository()
        self.pathsProvider = PathsProvider()
        self.attachmentTypes = ImageAttachmentTypes()
        self.detectionTypes = DetectionTypes()
        self.resultsRepository = DetectionResultRepository()
        self.logger = LoggerFactory()

    @exception
    def prepare_and_upload_results(self, request_id: int, results,
                                   image_file_path):
        for res in results:
            type_name = res[0]
            file_name = f"{type_name}.jpg"
            faces = res[1]
            processing_time = res[2]
            if faces is None:
                return
            self.logger.info(f"Working on results for {type_name} \n{faces}")
            result_file_path = os.path.join(
                self.pathsProvider.local_detection_image_path(),
                str(request_id), file_name)
            self.__save_result_image_to_local_directory__(
                faces, image_file_path, result_file_path)
            result_entity = self.__prepare_result_entities__(
                faces, file_name, request_id, type_name, processing_time)
            result_id = self.resultsRepository.add_detection_result_with_image(
                result_entity)
            result_file = open(result_file_path, "rb")
            self.filesUploader.upload_detection_result(result_id,
                                                       result_file.read(),
                                                       file_name)

    def __prepare_result_entities__(self, faces, file_name, request_id,
                                    type_name, proc_time):
        image_attachment = ImageAttachment(
            file_name, self.attachmentTypes.detection_result_id)
        faces_coordinates = [DetectionRectangle(faces) for faces in faces]
        result_entity = DetectionResult(
            request_id, self.detectionTypes.get_type_id(type_name),
            image_attachment, faces_coordinates, str(proc_time))
        return result_entity

    def __save_result_image_to_local_directory__(self, faces, image_file_path,
                                                 result_file_path):
        image = cv2.imread(image_file_path)
        result_image_data = self.imageEditor.draw_faces(image, faces)
        cv2.imwrite(result_file_path, result_image_data)
コード例 #14
0
class NeuralNetworkResultsSaver():
    def __init__(self):
        self.logger = LoggerFactory()
        self.pathsProvider = PathsProvider()
        self.filesUploader = FilesUploader()
        self.nnFilesRepo = NeuralNetworkFileRepository()
        self.nnTypes = NeuralNetworkTypes()
        self.stringOperator = StringOperator()

    def save_result_files(self, neural_network_id, training_times,
                          data_preparation_time):
        base_path = path.join(self.pathsProvider.local_neural_network_path(),
                              str(neural_network_id))
        file_paths = [path.join(base_path, f) for f in listdir(base_path)]
        for file_path in file_paths:
            file_name, nn_type_id = self.__get_file_name_and_file_type_id(
                file_path)
            training_time_of_nn = training_times[nn_type_id]
            self.__save_data__(file_name, neural_network_id, nn_type_id,
                               file_path,
                               training_time_of_nn + data_preparation_time,
                               training_time_of_nn)
            # commented because files are now too big
            # self.__upload_result_file__(file_name, file_path, neural_network_id)

    def __save_data__(self, file_name, neural_network_id, nn_type_id,
                      file_path, process_time, training_time):
        file_size = get_file_size(file_path)
        neural_network_file_entity = NeuralNetworkFile(file_name,
                                                       neural_network_id,
                                                       nn_type_id,
                                                       str(process_time),
                                                       str(training_time),
                                                       file_size)
        self.nnFilesRepo.add_neural_network_file(neural_network_file_entity)
        self.logger.info(f"Upload of file {file_name} FINISHED")

    def __upload_result_file__(self, file_name, file_path, neural_network_id):
        self.logger.info(
            f"Upload of file {file_name} STARTED (possible timeout error on weak network and big file size)"
        )
        opened_file = open(file_path, 'rb')
        self.filesUploader.upload_neural_network(neural_network_id,
                                                 opened_file.read(), file_name)

    def __get_file_name_and_file_type_id(self, file_path):
        file_name = self.stringOperator.get_file_name_from_path(file_path)
        nn_type_name = self.stringOperator.find_between(file_name, '_', '.')
        nn_type_id = self.nnTypes.get_type_id(nn_type_name)
        return file_name, nn_type_id
コード例 #15
0
 def __init__(self):
     self.logger = LoggerFactory()
     self.neuralNetworkPersonRepo = NeuralNetworkPersonRepository()
     self.peopleImagesProvider = PeopleImagesProvider()
     self.neuralNetworkRepo = NeuralNetworkRepository()
     self.azureGroupTrainer = AzureLargeGroupTrainer()
     self.openCvNnTrainer = OpenCvNnTrainer()
コード例 #16
0
class FilesDownloader():
    def __init__(self):
        self.pathsProvider = PathsProvider()
        self.dropbox = DropboxClient()
        self.directory = DirectoryManager()
        self.logger = LoggerFactory()

    def download_detection_input(self, request_id: int):
        dropbox_request_path = os.path.join(
            self.pathsProvider.dropbox_detection_image_path(),
            str(request_id)).replace("\\", "/")
        local_save_path = os.path.join(
            self.pathsProvider.local_detection_image_path(), str(request_id))
        self.__single_file_download__(dropbox_request_path, local_save_path)
        self.logger.info(
            F"Finished downloading input for detection with id :{request_id}")

    def download_person(self, person_id: int):
        dropbox_person_path = os.path.join(
            self.pathsProvider.dropbox_person_image_path(),
            str(person_id)).replace("\\", "/")
        local_save_path = os.path.join(
            self.pathsProvider.local_person_image_path(), str(person_id))
        self.__multiple_file_download__(dropbox_person_path, local_save_path)
        self.logger.info(F"Finished downloading person with id :{person_id}")

    def download_recognition_input(self, request_id: int):
        dropbox_request_path = os.path.join(
            self.pathsProvider.dropbox_recognition_image_path(),
            str(request_id)).replace("\\", "/")
        local_save_path = os.path.join(
            self.pathsProvider.local_recognition_image_path(), str(request_id))
        self.__single_file_download__(dropbox_request_path, local_save_path)
        self.logger.info(
            F"Finished downloading input for recognition with id :{request_id}"
        )

    def download_neural_network(self, nn_id: int):
        dropbox_neural_network_path = os.path.join(
            self.pathsProvider.dropbox_neural_network_path(),
            str(nn_id)).replace("\\", "/")
        local_save_path = os.path.join(
            self.pathsProvider.local_neural_network_path(), str(nn_id))
        self.__multiple_file_download__(dropbox_neural_network_path,
                                        local_save_path)
        self.logger.info(
            F"Finished downloading neural network with id :{nn_id}")

    def __single_file_download__(self, dropbox_request_path, local_save_path):
        self.directory.create_directory_if_doesnt_exist(local_save_path)
        self.dropbox.download_single_file(dropbox_request_path,
                                          local_save_path)

    def __multiple_file_download__(self, dropbox_folder_path, local_save_path):
        self.directory.clean_directory(local_save_path)
        self.directory.create_directory_if_doesnt_exist(local_save_path)
        self.dropbox.download_folder(dropbox_folder_path, local_save_path)
コード例 #17
0
    def __init__(self):
        self.logger = LoggerFactory()
        gpio.setmode(gpio.BCM)

        def read_pin(self, number: int):

            gpio.setup(23, gpio.IN, pull_up_down=gpio.PUD_DOWN)
        gpio.read()
コード例 #18
0
 def __init__(self):
     self.logger = LoggerFactory()
     self.faceRecognizerProvider = FaceRecognizerProvider()
     self.openCvFaceRecognizer = OpenCvFaceRecognizer()
     self.azureFaceRecognizer = AzureFaceRecognizer()
     self.azureNnClient = AzureLargeGroupsClient()
     self.azureFaceClient = AzureFaceClient()
     self.neuralNetworkFilesRepo = NeuralNetworkFileRepository()
     self.recognitionResultRepo = RecognitionResultRepository()
コード例 #19
0
class MovementResultOperator:
    def __init__(self):
        self.imageEditor = ImageEditor()
        self.pathsProvider = PathsProvider()
        self.attachmentTypes = ImageAttachmentTypes()
        self.filesUploader = FilesUploader()
        self.movementRepo = MovementRepository()
        self.directoryManager = DirectoryManager()
        self.logger = LoggerFactory()

    def prepare_and_upload_result(self, frame, movements):
        self.logger.info("Movement detected")
        file_name = f"motion_{datetime.now().date()}.png"
        new_id = self.__add_movement_to_db__(file_name)
        file_path_to_create = os.path.join(
            self.pathsProvider.local_motion_image_path(), str(new_id))
        file_path = os.path.join(file_path_to_create, file_name)
        self.directoryManager.create_directory_if_doesnt_exist(
            file_path_to_create)
        self.__save_result_image_to_local_directory__(frame, movements,
                                                      file_path)
        self.__upload_movement_photo__(file_name, file_path, new_id)

    def __upload_movement_photo__(self, file_name, file_path, new_id):
        self.logger.info(f"Uploading file :{file_path}")
        result_file = open(file_path, "rb")
        self.filesUploader.upload_detected_motion(new_id, result_file.read(),
                                                  file_name)

    def __add_movement_to_db__(self, file_name):
        image_attachment = ImageAttachment(file_name,
                                           self.attachmentTypes.movement_id)
        notification_entity = Movement("Movement detected", image_attachment)
        motion_id = self.movementRepo.add_movement(notification_entity)
        return motion_id

    def __save_result_image_to_local_directory__(self, frame_to_upload,
                                                 detected_movemenets,
                                                 file_name):
        if MOVEMENT_TIMESTAMP:
            self.imageEditor.mark_frame(frame_to_upload, 'Occupied')
        if MOVEMENT_MARKING:
            self.imageEditor.draw_contour(detected_movemenets, frame_to_upload)
        cv2.imwrite(file_name, frame_to_upload)
コード例 #20
0
class PeopleDownloader():
    def __init__(self):
        self.logger = LoggerFactory()
        self.pathProvider = PathsProvider()
        self.directoryManager = DirectoryManager()
        self.peopleRepo = PersonRepository()
        self.filesDownloader = FilesDownloader()

    def download_people_to_local(self):
        people_path = self.pathProvider.local_person_image_path()
        directories = self.directoryManager.get_subdirectories_with_files_count(
            people_path)
        people = self.peopleRepo.get_people_ids_with_images_count()
        people_to_download = [x for x in people if x not in directories]
        self.logger.info(f"directories {directories} "
                         f"\npeople: {people}"
                         f"\npeople_to_download: {people_to_download}")
        for p in people_to_download:
            self.filesDownloader.download_person((p[0]))
コード例 #21
0
class NeuralNetworksProvider():
    def __init__(self):
        self.pathProvider = PathsProvider()
        self.logger = LoggerFactory()
        self.filesDownloader = FilesDownloader()
        self.nnRepo = NeuralNetworkRepository()
        self.directoryManager = DirectoryManager()

    def download_neural_networks_to_local(self):
        nn_path = self.pathProvider.local_neural_network_path()
        directories = self.directoryManager.get_subdirectories_with_files_count(
            nn_path)
        neural_networks = self.nnRepo.get_completed_neural_networks_ids_with_downloadable_files_count(
        )
        nns_to_download = [x for x in neural_networks if x not in directories]
        self.logger.info(f"directories {directories} "
                         f"\nneural_networks: {neural_networks}"
                         f"\nneural_networks_to_download: {nns_to_download}")
        for nn in nns_to_download:
            self.filesDownloader.download_neural_network(nn[0])
コード例 #22
0
ファイル: tester.py プロジェクト: szymek325/MasterThesis
class Tester:
    def __init__(self):
        self.logger = LoggerFactory()
        self.imagesProvider = PeopleImagesProvider()
        self.recognizersProvider = FaceRecognizerProvider()
        self.openCvRecognizer = OpenCvFaceRecognizerForTesting()
        self.azureRecognizer = AzureFaceRecognizer()
        self.faceDetector = FaceDetectorsManager()
        self.neuralNetworkToUse = 40

    def test_all_photos(self):
        faces = 0
        all_photos_with_people = self.imagesProvider.get_image_paths_for_people(range(2, 102), 20)
        self.logger.info("test")
        for person_id, person_image in all_photos_with_people:
            found_faces= self.faceDetector.get_face_by_azure(person_image)
            if found_faces == 0 or len(found_faces) == 0:
                continue
            else:
                faces = faces + 1
            self.logger.info(f"FACES: {faces} ")
コード例 #23
0
class FaceDetectionProcess():
    def __init__(self):
        self.logger = LoggerFactory()
        self.config = ConfigReader()
        self.request_manager = DetectionRequestsManager()
        self.directory = DirectoryManager()
        self.faceDetectionRepository = FaceDetectionRepository()
        self.pathsProvider = PathsProvider()

    @exception
    def run_face_detection(self):
        self.logger.info("  START FaceDetection")
        requests = self.faceDetectionRepository.get_all_not_completed()
        if not requests == null:
            for request in requests:
                try:
                    self.request_manager.process_request(request)
                    self.faceDetectionRepository.complete_request(request.id)
                except Exception as ex:
                    self.logger.error(
                        f"Exception when processing detection {request.id}.\n Error: {str(ex)}"
                    )
                    self.faceDetectionRepository.complete_with_error(
                        request.id)
            self.directory.clean_directory(
                self.pathsProvider.local_detection_image_path())
        self.logger.info("  END FaceDetection")
コード例 #24
0
class OpenCvFaceRecognizerForTesting():
    def __init__(self):
        self.logger = LoggerFactory()
        self.faceDetectorManager = FaceDetectorsManager()
        self.imageConverter = ImageConverter()

    def recognize_with_single_recognizer_with_haar(self, face_recognizer,
                                                   image_path):
        detection_start = time.time()
        image = cv2.imread(image_path)
        detected_faces = self.faceDetectorManager.get_face_by_haar(image)
        detection_end = time.time()
        detection_time = detection_end - detection_start
        start_time = time.time()
        self.logger.info(f"Using {face_recognizer} recognizer on {image_path}")
        if len(detected_faces) is 0:
            return 0
        (startX, startY, endX, endY) = detected_faces[0]
        predict_image = self.imageConverter.convert_to_np_array(
            image[startY:endY, startX:endX])
        nbr_predicted, confidence = face_recognizer.predict(predict_image)
        return nbr_predicted

    def recognize_with_single_recognizer_with_dnn(self, face_recognizer,
                                                  image_path):
        detection_start = time.time()
        image = cv2.imread(image_path)
        detected_faces = self.faceDetectorManager.get_face_by_dnn(image)
        detection_end = time.time()
        detection_time = detection_end - detection_start
        start_time = time.time()
        self.logger.info(f"Using {face_recognizer} recognizer on {image_path}")
        if len(detected_faces) is 0:
            return 0
        (startX, startY, endX, endY) = detected_faces[0]
        predict_image = self.imageConverter.convert_to_np_array(
            image[startY:endY, startX:endX])
        nbr_predicted, confidence = face_recognizer.predict(predict_image)
        return nbr_predicted
コード例 #25
0
class Dht11Reader:
    def __init__(self):
        self.logger = LoggerFactory()
        gpio.setmode(gpio.BCM)

    def get_readings(self):
        self.logger.info(f"Reading value from dht11")
        try:
            self.logger.info(f"read value here")
        except Exception as ex:
            self.logger.error(f"Error '{ex}' when reading value from DHT11")
コード例 #26
0
class Tester:
    def __init__(self):
        self.logger = LoggerFactory()
        self.imagesProvider = PeopleImagesProvider()
        self.recognizersProvider = FaceRecognizerProvider()
        self.openCvRecognizer = OpenCvFaceRecognizerForTesting()
        self.azureRecognizer = AzureFaceRecognizer()
        self.faceDetector = FaceDetectorsManager()
        self.neuralNetworkToUse = 44

    def test_all_photos(self):
        azure_counter = 0
        fisher_counter = 0
        eigen_counter = 0
        lbph_counter = 0
        faces = 0
        all_photos_with_people = self.imagesProvider.get_image_paths_for_people(
            range(2, 102), 20)
        fisher, eigen, lbph = self.recognizersProvider.create_open_cv_face_recognizers_with_type(
            self.neuralNetworkToUse)
        self.logger.info("test")
        for person_id, person_image in all_photos_with_people:
            faces = faces + 1
            azure_result = 0
            fisher_result = 0
            eigen_result = 0
            lbph_result = 0
            try:
                azure_result = self.azureRecognizer.recognize_face_without_db(
                    18, person_image)
                fisher_result = self.openCvRecognizer.recognize_with_single_recognizer_with_dnn(
                    fisher, person_image)
                eigen_result = self.openCvRecognizer.recognize_with_single_recognizer_with_dnn(
                    eigen, person_image)
                lbph_result = self.openCvRecognizer.recognize_with_single_recognizer_with_dnn(
                    lbph, person_image)
            except Exception as exception:
                self.logger.error(exception)
            if person_id == int(azure_result):
                azure_counter = azure_counter + 1
            if person_id == fisher_result:
                fisher_counter = fisher_counter + 1
            if person_id == eigen_result:
                eigen_counter = eigen_counter + 1
            if person_id == lbph_result:
                lbph_counter = lbph_counter + 1
            self.logger.info(
                f"FACES: {faces} Azure: {azure_counter} Fisher: {fisher_counter}"
                f" Eigen: {eigen_counter} LBPH: {lbph_counter}")
コード例 #27
0
class DetectionRequestsManager():
    def __init__(self):
        self.logger = LoggerFactory()
        self.faceDetectorsManager = FaceDetectorsManager()
        self.resultsOperator = ResultsOperator()
        self.inputFileProvider = InputFileProvider()

    def process_request(self, request: Detection):
        self.logger.info(
            f"Working on Face Detection Request id: {request.id} started")
        input_file_path = self.inputFileProvider.get_detection_input_file_path(
            request.id)
        self.logger.info(f"Running all face detectors")
        results = self.faceDetectorsManager.get_faces_on_image_from_file_path(
            input_file_path)
        self.resultsOperator.prepare_and_upload_results(
            request.id, results, input_file_path)
        self.logger.info(f"Finished Face Detection Request id: {request.id} ")
コード例 #28
0
class NeuralNetworkTrainingRunner():
    def __init__(self):
        self.logger = LoggerFactory()
        self.peopleManager = PeopleDownloader()
        self.nnManager = NeuralNetworkRequestsManager()
        self.nnRepo = NeuralNetworkRepository()

    @exception
    def run_training(self):
        self.logger.info("  START NeuralNetworkTraining")
        requests = self.nnRepo.get_all_not_completed()
        if not requests == null and requests.count() is not 0:
            self.peopleManager.download_people_to_local()
            for request in requests:
                try:
                    self.nnManager.process_request(request)
                except Exception as ex:
                    self.logger.error(f"Exception when processing neural network training {request.id}.\n Error: {str(ex)}")
                    self.nnRepo.complete_with_error(request.id)
        self.logger.info("  END NeuralNetworkTraining")
コード例 #29
0
class SensorsOperator():
    def __init__(self):
        self.logger = LoggerFactory()
        self.readingsRepo = SensorsReadingRepository()
        self.dht11Reader = Dht11ReaderCheater()
        self.notifications = NotificationsService()
        self.settingsTypes = NotificationSettingsTypes()

    def add_readings(self):
        self.logger.info(f"Starting retriving readings")
        temp, hum = self.dht11Reader.read_values()
        self.logger.info(f"Temp: {temp}\n Hum: {hum}")
        self.readingsRepo.add_reading(temp, hum)
        readings_dictionary = {
            self.settingsTypes.temperature: temp,
            self.settingsTypes.humidity: hum
        }
        self.notifications.add_reading_notification_if_required(
            readings_dictionary)
        self.logger.info(f"Sensors readings finished")
コード例 #30
0
class FaceRecognitionProcess():
    def __init__(self):
        self.logger = LoggerFactory()
        self.config = ConfigReader()
        self.faceRecognitionManager = RecognitionRequestManager()
        self.faceRecognitionRepo = FaceRecognitionRepository()
        self.neuralNetworksManager = NeuralNetworksProvider()

    @exception
    def run_face_recognition(self):
        self.logger.info("  START FaceRecognition")
        requests = self.faceRecognitionRepo.get_all_not_completed()
        if not requests == null and requests.count() is not 0:
            self.neuralNetworksManager.download_neural_networks_to_local()
            for request in requests:
                try:
                    self.faceRecognitionManager.process_request(request)
                    self.faceRecognitionRepo.complete_request(request.id)
                except Exception as ex:
                    self.logger.error(
                        f"Exception when processing recognition {request.id}.\n Error: {str(ex)}"
                    )
                    self.faceRecognitionRepo.complete_with_error(request.id)
        self.logger.info("  END FaceRecognition")