my-caffe2-experiments

This repository is my experiments of Caffe2 framework for my practise.

1. The first example include implementation of LeNet network for MNIST database. Link
2. CIFAR 10 caffe2 training and inference. Link
3. Caffe2 Utils - These are the common functions that are used for general training and testing. Also, methods are included to create a database in minidb. (Note: Some of the methods are taken from caffe2 tutorials.)
4. Caffe2 Code Generator - This is a work in progress tool that takes a trained caffe prototxt file and generates the caffe2 code that can be used for training and inference on caffe2.