Update prediction.py
Browse files- prediction.py +1 -5
prediction.py
CHANGED
@@ -19,16 +19,12 @@ class smartcities:
|
|
19 |
self.detect_fn = detect_fn
|
20 |
|
21 |
def predict(self, input):
|
22 |
-
# Ruta del video (Se debe cargar de manera manual)
|
23 |
-
# PATH_VIDEO = "/tmp/in_video.mp4"
|
24 |
-
# video_result = open(PATH_VIDEO, "wb")
|
25 |
-
# video_result.write(base64.b64decode(image_64_decode))
|
26 |
|
27 |
# Ruta del video en donde almacenaremos los resultados
|
28 |
PATH_OUTPUT = "video_out.mp4"
|
29 |
|
30 |
# Cuántos frames vamos a saltarnos (Durante estos frames nuestro algoritmo de seguimiento funciona)
|
31 |
-
SKIP_FPS =
|
32 |
|
33 |
# Cuál será el umbral mínimo par que se considere una detección
|
34 |
TRESHOLD = 0.5
|
|
|
19 |
self.detect_fn = detect_fn
|
20 |
|
21 |
def predict(self, input):
|
|
|
|
|
|
|
|
|
22 |
|
23 |
# Ruta del video en donde almacenaremos los resultados
|
24 |
PATH_OUTPUT = "video_out.mp4"
|
25 |
|
26 |
# Cuántos frames vamos a saltarnos (Durante estos frames nuestro algoritmo de seguimiento funciona)
|
27 |
+
SKIP_FPS = 30
|
28 |
|
29 |
# Cuál será el umbral mínimo par que se considere una detección
|
30 |
TRESHOLD = 0.5
|