lenet_mnist / README.md
rzimmerdev's picture
Added dataset suggestions for deciding which model to use.
b8ee4fc
|
raw
history blame
656 Bytes
# zero-to-hero
Create and deploy to production a simple neural network for Computer Vision
# Tools Used
* JAX Library for computing gradients, performing tensor operations and scheming the segmentation model
* Wandb for metrics and training tools
* MLflow for deploying and compiling the model for production
* Gradio for interactive user-experience platform within an online platform (Data-ICMC Website).
# Datasets to consider
* [LabelMe 12 50k](https://www.kaggle.com/datasets/dschettler8845/labelme-12-50k)
* [City-Scapes](https://www.cityscapes-dataset.com/dataset-overview/)
* [CamVid](http://mi.eng.cam.ac.uk/research/projects/VideoRec/CamVid/)