Update processar_imagem_de_entrada.py
Browse files
processar_imagem_de_entrada.py
CHANGED
@@ -5,6 +5,9 @@ import numpy as np
|
|
5 |
import cv2
|
6 |
|
7 |
# Função para processar a imagem de entrada
|
|
|
|
|
|
|
8 |
# Função para processar a imagem de entrada
|
9 |
def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
|
10 |
try:
|
@@ -40,14 +43,15 @@ def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
|
|
40 |
# Converta o resultado para um formato booleano
|
41 |
logit = logit.astype(bool)
|
42 |
|
43 |
-
|
44 |
# Salve o resultado em um arquivo se a opção estiver ativada
|
45 |
if salvar_resultado:
|
46 |
nome_arquivo = os.path.join(caminho_salvar_resultado, f'resultado_{int(time.time())}.png')
|
47 |
-
cv2.imwrite(nome_arquivo, logit)
|
48 |
-
|
49 |
-
return spent, logit
|
50 |
|
51 |
except Exception as e:
|
52 |
# Em caso de erro, retorne uma mensagem de erro
|
|
|
53 |
return str(e), None
|
|
|
|
5 |
import cv2
|
6 |
|
7 |
# Função para processar a imagem de entrada
|
8 |
+
import logging
|
9 |
+
logging.basicConfig(level=logging.DEBUG) # Enable logging
|
10 |
+
|
11 |
# Função para processar a imagem de entrada
|
12 |
def processar_imagem_de_entrada(img, modelo, pipe, salvar_resultado=False):
|
13 |
try:
|
|
|
43 |
# Converta o resultado para um formato booleano
|
44 |
logit = logit.astype(bool)
|
45 |
|
|
|
46 |
# Salve o resultado em um arquivo se a opção estiver ativada
|
47 |
if salvar_resultado:
|
48 |
nome_arquivo = os.path.join(caminho_salvar_resultado, f'resultado_{int(time.time())}.png')
|
49 |
+
cv2.imwrite(nome_arquivo, logit)
|
50 |
+
|
51 |
+
return spent, logit
|
52 |
|
53 |
except Exception as e:
|
54 |
# Em caso de erro, retorne uma mensagem de erro
|
55 |
+
logging.error(f"Error: {str(e)}") # Log the error message
|
56 |
return str(e), None
|
57 |
+
|