Skip to content

Instantly share code, notes, and snippets.

@nestorandrespe
Last active September 9, 2021 23:57
Show Gist options
  • Save nestorandrespe/6dfad634fcf8fda29d3e3991e68d26f1 to your computer and use it in GitHub Desktop.
Save nestorandrespe/6dfad634fcf8fda29d3e3991e68d26f1 to your computer and use it in GitHub Desktop.
Paso a paso para entrenar y exportar un modelo Pix2pix

Instrucciones para usar Pix2Pix con tensorflow

Clonar el repositorio

Descargar o clonar el repositorio en github

Entrenar un modelo usando un dataset específico

Abrir la terminal dentro de la carpeta del Pix2pix y desde ahi ejecutar los siguientes comandos:

python pix2pix.py \
--mode train \ # modo entrenamiento
--input_dir [carpeta] \ # carpeta con el dataset
--output_dir [carpeta] \ # carpeta donde se va a guardar el modelo
--which_direction [AtoB o BtoA] \ # sentido de la imagen
--max_epochs [números] \ #número de generaciones
--checkpoint [carpeta output_dir] \ # en caso de retomar el entrenamiento
--aspect_ratio [ratio de la imagen] # por defecto es 1

Exportar un modelo

python pix2pix.py \
--mode export \ # modo exportar
--output_dir export \ # carpeta export al interior de la carpeta raíz
--checkpoint [carpeta] \ # carpeta output_dir del entrenamiento
--which_direction [AtoB o BtoA] # sentido de la imagen
cd server
python tools/export-checkpoint.py \
--checkpoint ../export \
--output_file [archivo] # archivo *.pict
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment