Week10- StyleTransfer Training

Traning

The training script for StyleTransfer depends on a quite old TensorFlow version which imports extra difficulties for building the environment. Moreover, the training process uses the CoCo dataset which contains 14 GB of images. It also introduces a lot of training time. 

In my case, I used RTX2080Ti to do the training process. For two epochs and 8 batch, it takes about 3 hours to finish the training phase. In general, it is super fast considering how many images CoCo could have.

Convert

Since this is an old script, I can not directly transform it into a browser compatible model. In this case,  I used the ml5 script to do the work. 

Testing

Below aresome images I converted:

Leave a Reply