StylePY icon indicating copy to clipboard operation
StylePY copied to clipboard

Neural style transfer project with Flask

StylePY

  • With this web application, you can transfer styles between your images and videos in a simple interface.

  • You will be able to create magnificent pictures with Neural style transfer.

  • If I explain the purpose of the application in a simple way, we can say that it is to transfer the style of a picture to another picture, or we can say to transfer the style of a picture to another video.

  • web application was written in flask, neural style transfer was written in pytorch.

  • This application requires high GPU usage, if you do not have a good GPU, read my article titled Deploying flask project over Colab which I wrote earlier. https://github.com/AhmetFurkanDEMIR/Unix-and-Cloud-Computing/blob/master/GogleColab_Flask/README.md

  • Usage video of the application : https://www.linkedin.com/posts/1dfurkan_neuralstyletransfer-computervision-machinelearning-activity-6717726740375969792-RkTN

Use of the application

  • First, go to the folder named models and run the code. (Since the size of the model (500 mb) is very large, you will download the model remotely with this code.) code :
bash ./download_models.sh
  • Now let's install the modules required for the operation of this web application with pip. code :
pip3 install -r requirements.txt
  • Now everything is ok, we can run the application :). code :
python3 main.py
  • If you see the picture below, everything is fine, proceed to the link in the window that opens and enjoy the application. Screenshot_2020-10-02_11-32-26

Application images

Screenshot_20201002-114542_Chrome

Screenshot_2020-10-02_11-41-41

Screenshot_2020-10-02_11-41-59

Screenshot_2020-10-02_11-43-03

Screenshot_2020-10-02_11-43-25

Test results

  • Style Image

out1

out1(1)

  • Style Video

asdddd-min