main
: Pipeline for training and evaluationconfig
: Model config and file pathscheckpoint
: MXNet checkpoint filesdata
: Rawdata and temporal dataPretrain_model
: pretrained model filemodel
: MXNet Symbols of VGG16, Resnetutils.preprocessing
: Utility function for preprocessing, such as PCA, Rescaling.utils.io
: Utility function for I/Outils.filter
: Weighted median filterutils.label_data
: A handy tool for labeling data manually.
- Apache MXNet 0.12.0
- python 2.7
python main.py --model vgg16 --opt sgd --lr 1e-5 --t1 0.2 --t2 0.99 --epoch 0 --num_epoch 100