zaina-ml/ml_forge
A visual-based graph node editor for training computer vision models.
This tool helps researchers, data scientists, and machine learning engineers create, train, and evaluate image classification models without writing code. You provide image datasets (like MNIST, CIFAR, or your own images), define your model visually by connecting building blocks, and then train it. The output is a trained computer vision model and the underlying PyTorch code.
365 stars.
Use this if you need to quickly prototype or train image classification models and prefer a visual, drag-and-drop interface over writing Python code.
Not ideal if you primarily work with non-image data (like text or tabular data) or require advanced, custom machine learning architectures beyond image classification.
Stars
365
Forks
44
Language
Python
License
MIT
Category
Last pushed
Mar 25, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/computer-vision/zaina-ml/ml_forge"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related tools
lucidrains/vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with...
roflcoopter/viseron
Self-hosted, local only NVR and AI Computer Vision software. With features such as object...
blakeblackshear/frigate
NVR with realtime local object detection for IP cameras
levan92/deep_sort_realtime
A really more real-time adaptation of deep sort
notAI-tech/NudeNet
Lightweight nudity detection