The computer-vision from tiagonuneslx

Aprendizagem Profunda para Visão por Computador

Grupo 8

computer-vision's People

Contributors

Importar o modelo.
Carregar algumas imagens (para já vou buscar ao test set da wider_face)
Dar parse das anotações das imagens
Dar predict das imagens com o modelo.
Ajustar as bounding boxes para as imagens do OpenCV
Criar a lógica para aumentar o tamanho das bounding boxes.
Encontrar a cara com métodos tradicionais de CV e criar uma máscara.
Aplicar a máscara sobre as imagens (para já vou deixá-la preta).
Mostrar os resultados.
Calcular Precision, Recall, Avg IoU.
Fazer o testing do modelo que faltou fazer, com o test dataset do wider_face.
Ignorar as ground truth bounding boxes demasiado pequenas do training set, o YOLO nunca vai conseguir encontrá-las e está a estragar um bocado o cálculo da loss.
Fazer test do model
Mostrar gráficos de loss do model
Arranjar o callback do modelo que calcula o mAP a cada iteração de training, e guarda o melhor modelo.
Aumentar o número de iterações (EPOCH = 5), e correr o modelo (vai demorar bué).
Dar tweaks nos hyperparameters.