our first ai.
/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
yum install git
git clone https://github.com/yyuu/pyenv.git ~/.pyenv
echo 'export PYENV_ROOT="$HOME/.pyenv"' >> ~/.bash_profile
echo 'export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bash_profile
echo 'eval "$(pyenv init -)"' >> ~/.bash_profile
pyenv install 3.5.2
pyenv global 3.5.2
pip install keras==1.2.2
pip install tensorflow
pip install pillow
pip install h5py
brew install graphviz
pip install pydot
pip install pydot-ng
pip install pyparsing
pip install matplotlib
- completely train by yourself
cd catvsdog/01_job
python scratch.py
- use vgg16 base, train top fc only.
cd catvsdog/01_job
python bottleneck.py
- use vgg16 base, train top fc and the 16th convenience2D.
cd catvsdog/01_job
python finetuning.py
you can set mode from 1 to 3(1: scratch, 2: bottleneck, 3: finetuning)
and use image param to specify an image.
for example:
cd catvsdog/01_job
python predict.py --mode 3 --image ../99_data/input/1.png