Skip to content

Built an interactive deep learning app with Streamlit and PyTorch to apply for style transfer.

Notifications You must be signed in to change notification settings

adilsonmedronha/style-transfer

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Desenvolvimento para o Trabalho II da disciplina de Aprendizado de Máquina, PUCRS.

train

python src/neural_style.py train --dataset --style-image images/style_to_train/sketch1.jpg --save-model-dir models/ --epochs 2 --cuda 0

eval

streamlit run src/main.py

About

Built an interactive deep learning app with Streamlit and PyTorch to apply for style transfer.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%