Details
This demo combines style transfer models (please see the details section of my
Real-time style transfer demo for information on how the models were optimised
using techniques such as pruning) and
a semantic segmentation model to extract people out of an image. The semantic segmentation model is based on
MobileNetV3 and was mostly trained on open datasets and lightly pruned to give the final model.
All models were trained and optimised in Pytorch and then converted for deployment on TensorFlow.js.
For deployment, all the layers in the models that could be fused were fused and the models quantised
to give the final optimised models.
Currently working on
-
Deploying arbitrary style transfer model to allow for style transfer from any user defined image
-
Making a video version of the demo
-
Deploying the demo as a React Native app
Credits
This demo was put together by Zawar Qureshi
but could not have been done without the following:
I'd love to hear from people interested
in making tools/apps using these models!