DHEIVER commited on
Commit
c5a7258
1 Parent(s): 70c73be

Update processar_imagem_de_entrada.py

Browse files
Files changed (1) hide show
  1. processar_imagem_de_entrada.py +8 -4
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) # Corrected variable name
48
-
49
- return spent, logit # Corrected variable name
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
+