giorg0s / 2020-tfg-jorge_guerrero

0 stars 2 forks source link

EXPERIMENTO - Estudiar cómo y pasar una red entrenada en PyTorch a TensorFlow y TensorFlow Lite #6

Open giorg0s opened 4 years ago

giorg0s commented 4 years ago

Según comenta Andrés los pasos a seguir podrían ser:

  1. Convertir Resnet pre-entrenada.
  2. Convertir MovilenetV2 pre-entrenada.
  3. Convertir una deconvolucional sin entrenar.

La conversión idónea debería ser de Pytorch 1.6 a TF 1.14-15 o TF 2.

giorg0s commented 4 years ago

https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html

giorg0s commented 4 years ago

https://medium.com/@santhoshkolloju/convert-your-pytorch-models-to-tensorflow-with-onnx-84c3bdd8d722

giorg0s commented 4 years ago

Al acabar el proceso PyTorch -> ONNX -> TF y visualizar el grafo resultante con tensorboard lo que obtengo es esto (concretamente usando un modelo de mobilenet_v2 pre-entrenado): image Creo que no tiene mucha pinta de haberlo hecho bien.

jmbuena commented 4 years ago

Parece algo raro, sí. Lo suyo sería mirar la arquitectura de red de mobilenet_v2 ya implementada en tensorflow y ver como es.

giorg0s commented 4 years ago

Parece algo raro, sí. Lo suyo sería mirar la arquitectura de red de mobilenet_v2 ya implementada en tensorflow y ver como es.

Sí, creo que el problema está cuando paso el modelo de ONNX a tensorflow, porque al visualizarlo tiene sentido:

mobilenet_v2_onnx