DHEIVER commited on
Commit
2aa8a22
1 Parent(s): aa1c905

Create detect_anomalies.py

Browse files
Files changed (1) hide show
  1. detect_anomalies.py +13 -0
detect_anomalies.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import numpy as np
2
+ import cv2
3
+
4
+ def detect_anomalies(mask):
5
+ # Aplicar um limiar à máscara para identificar as regiões de anomalia
6
+ threshold = 0.5 # Ajuste o limiar conforme necessário
7
+ anomalies = (mask > threshold).astype(np.uint8)
8
+
9
+ # Aplicar pós-processamento, como erosão e dilatação, se necessário
10
+ # anomalies = cv2.erode(anomalies, kernel, iterations=1)
11
+ # anomalies = cv2.dilate(anomalies, kernel, iterations=1)
12
+
13
+ return anomalies