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:
