working in progress
git clone --recursive https://github.com/legokichi/keras-segnet.git
pyenv shell anaconda3-4.1.1
sudo apt-get install graphviz
conda install theano pygpu
pip install tensorflow-gpu
pip install keras
pip install mypy
pip install pydot_ng
pip install imgaug
mypy --ignore-missing-imports train.py
python model_segnet.py
source download_camvid.sh
source download_mscoco.sh
env CUDA_VISIBLE_DEVICES=0 python train.py
env CUDA_VISIBLE_DEVICES=1 python train.py --indices
env CUDA_VISIBLE_DEVICES=0 python train.py --coco
env CUDA_VISIBLE_DEVICES=0 python train.py --coco --ker_init=he_normal --lr=0.001 --optimizer=nesterov
tensorboard --port=8888 --logdir=log
jupyter notebook --ip=0.0.0.0
If learning does not start, memory is not enough so please change MultiprocessIterator to SerialIterator in CamVid.py or mscoco.py.
env CUDA_VISIBLE_DEVICES=0 python train.py --initial_epoch=5 --resume=2017-04-17-08-29-19_weights.epoch0005-val_loss0.43.hdf5
env CUDA_VISIBLE_DEVICES=1 python train.py --initial_epoch=5 --resume=2017-04-17-08-31-47_indices_weights.epoch0005-val_loss0.48.hdf5 --indices
working in progress