A tool to generate PyTorch neural network code from flowchart images
Project description
Sketch NN
Transform hand-drawn neural network sketches into functional PyTorch code
Sketch NN is an innovative Python library that transforms hand-drawn neural network sketches into functional PyTorch code. Design your neural architectures on paper, capture them with our tool, and watch as Sketch NN brings your ideas to life!
🌟 Features
- 📸 Process hand-drawn or digital sketches of neural network architectures
- 🧠 Support for a wide range of neural network layers
- 🔧 Generate ready-to-use PyTorch code
- 🖥️ User-friendly Gradio web interface for quick prototyping
- 🚀 FastAPI backend for scalable deployment
🛠️ Supported Layers
- Convolutional (Conv2D)
- Pooling (MaxPool2D, AvgPool2D)
- Fully Connected (Linear)
- Batch Normalization
- Dropout
- Activation Functions (ReLU, LeakyReLU, Sigmoid, Tanh)
- Recurrent (LSTM, GRU)
- Transformer
- Multi-head Attention
🚀 Installation
Install Sketch NN using pip:
pip install sketch_nn
🚀 Usage
from sketch_nn import NeuralNetworkDesigner
designer = NeuralNetworkDesigner()
designer.process_image('path_to_your_sketch.png')
pytorch_code = designer.generate_pytorch_code()
designer.write_to_file(pytorch_code, 'custom_nn.py')
Gradio Web Interface Demo
from sketch_nn.demo import run_gradio_demo
run_gradio_demo()
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
sketch_nn-0.1.5.tar.gz
(5.9 kB
view hashes)
Built Distribution
Close
Hashes for sketch_nn-0.1.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b868191dd01aa4a5a515ec5bb457ed29d84bed58ce2ad1a65f2a3b427f831faa |
|
MD5 | a5b88b41b26e9bd82824f9bca5e2e162 |
|
BLAKE2b-256 | bf44eb1014eb007b574bd9afd1bebc56806531e2b2e92e281493af9d0dde2a05 |