search

giorg0s / 2020-tfg-jorge_guerrero

0 stars 2 forks source link

Portar la red de Andrés #7

Open giorg0s opened 3 years ago

giorg0s commented 3 years ago

Usando el modelo proporcionado por Andrés, hecho en PyTorch realizar un script que lo transforme a TensorFlow Lite de manera que, en el siguiente paso del trabajo, se pueda integrar como un módulo de Mediapipe.

Aunque en el fondo el objetivo deseable es tener un script que acepte una red "cualquiera" y la transforme a TensorFlow Lite asegurando siempre la integridad del modelo, independientemente de las características del mismo.

¿Cómo se aseguraría esta integridad? Realizando tests unitarios en cada fase de la traducción (preferiblemente con unittest https://docs.python.org/3/library/unittest.html) para ir comprobando que va bien la cosa, así como regresiones.

giorg0s commented 3 years ago

Para probar el modelo lo que se ha comentado es escoger unas 5-10 imágenes no muy complicadas de la wflw (https://wywu.github.io/projects/LAB/WFLW.html) y asegurarse de que va bien. En este caso me las proporciona Andrés.