A collection of machine and deep learning models designed to run on mobile devices.
Models in this repository contain code and utilities for training models as well as converting them to mobile-friendly formats like Core ML, TensorFlow Mobile, and TensorFlow Lite.
Fritz AI helps you teach your applications how to see, hear, feel, think, and sense. Create ML-powered features in your mobile apps for both Android and iOS. Start with our ready-to-use feature APIs or connect and deploy your own custom models.
Sign up for an account on Fritz AI in order to get started with machine learning in your apps.
- Image Labeling: Label images based on their content.
- Object Detection: Localize and label objects in an image with a bounding box.
- Style Transfer: Transform images into works of art by transfering the style of one image onto the content of another.
- Image Segmentation: Semantic segmentation of images. Assign a value to each pixel of an image corresponding to the type of object it belongs to.
- Create ML Playgrounds: A series of playgrounds for training models with Apple's Create ML tool
Don't see the model you're looking for? Open an issue and let us know!
To see live demonstrations of these models running on-device, the Heartbeat App is available in both the App Store (source code) and Play Store (source code).
If you'd like to incorporate any of these models or versions you've trained into your own app, head over to Fritz. SDKs are available for both iOS and Android.
Additional, non-code resources for machine learning and AI.
- Mobile ML GitHub Repositories: A list of repos with machine learning models ready for mobile, organized by feature type.
- AI and ML Landscape: Our curated list of helpful products and services for AI and machine learning.
- AI and ML Newsletters: A list of relevant newsletters in AI and machine learning.
- Mobile Development Newsletters: A list of relevant newsletters in iOS, Android, React Native, and Cross Platform development.
- Data Science Newsletters: A list of relevant data science and data analytics newsletters.
- Facebook Groups for AI/ML, Mobile, and Data Science: A list of AI/ML, mobile dev, and data science Facebook communities.
Large files like model checkpoints, data, and archives of compiled code are managed via git lfs. You need to have Git LFS installed in order to download these files. Installation instructions are available here.
If you have Git LFS installed, large files will download automatically by default. This can take a while and require a good connection. To clone this repository without downloading the model checkpoints, you can run:
GIT_LFS_SKIP_SMUDGE=1 git clone ...
Heartbeat is a community of developers interested in the intersection of mobile and machine learning. Chat with us in Slack and stay up to date on the latest mobile ML news with our Newsletter.
For any questions or issues, you can:
- Submit an issue on this repo
- Go to our Help Center
- Message us directly in Slack