This repository provides training notebooks for Lens Studio templates powered with Snap ML. Each notebook allows to train a model which then can be brought into Lens Studio project.
Allows you to train Style Transfer model based on provided image.
- Notebook
- Dataset: COCO dataset
- Template Guide
Provides an example of binary classification
- Notebook
- Dataset: CelebA
- Template Guide
Provides an example of the car detection
- Notebook
- Dataset: COCO
- Template Guide
Provides an example of pizza segmentation
- Notebook
- Dataset: COCO
- Template Guide
Provides an example of training a model that classifies spectrogram images generated from audio.
- Notebook
- Dataset: SpeechCommands
- Template Guide
Demonstrates how to train and compress popular image-to-image networks like CycleGAN and Pix2Pix so that we could achieve real time performance on mobile devices.
- Notebook
- Template Guide (This model is compatible with Style Transfer Template)
Demonstrates how to train an image classification models with Keras and TFLite model maker and quantize them using TensorFlow
Demonstrates how to train a multi-object detection model using custom datasets annotated by ourselves with a step-by-step example on berries detection.
Educational walkthrough of how to train your own image classifier from scratch and making it SnapML compatible. This is intended for educational purposes only. Please review relevant dataset licenses prior to usage.
A license file is included with each folder project. The full license can be found here: https://lensstudio.snapchat.com/template-license