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.4.tar.gz
(5.7 kB
view hashes)
Built Distribution
Close
Hashes for sketch_nn-0.1.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d23b989d3d1e7f9e89183e996bcd607d2747206d82f6b89064e603410d3fdfb7 |
|
MD5 | 9d9ce464ef93b0f5ff5819c2bcca4451 |
|
BLAKE2b-256 | 142506b49375af2b97624af27580997a49eea65481e825a91a8281779f84f99c |