Draw a pytorch model

App to convert a diagram drawing into a pytorch model code

to start app run: streamlit run app.py

drawit.mp4

train model to convert sketch to clean diagram:

download didi dataset

run python3 preprocess.py to prepare folders

run python3 train.py to train model