No description, website, or topics provided.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
cloudml
dataset
images
js
logdir
models @ f505cec
scripts
trainer
.gitignore
.gitmodules
README.md
generate_dataset.sh
requirements.txt

README.md

shogi-recognizer

Image Resouces

Prerequisite

  • Python >= 3.6
    • Tensorflow >= 1.12
    • Beautiful Soup >= 4.6
    • Pillow >= 5.0

Training

$ pip install -r requirements.txt

Prepare dataset

$ ./generate_dataset.sh

Train

$ git submodule update --init
$ PYTHONPATH=./models/research/slim python ./trainer/task.py

Training on Cloud ML Engine

$ export BUCKET_NAME=<your bucket name>
$ ./cloudml/upload_data.sh
$ ./cloudml/run.sh

Freeze graph

$ python ./scripts/freeze_graph.py --labels logdir/labels.txt --checkpoint_path logdir/model.ckpt --output_graph output_graph.pb

Convert to JS

$ tensorflowjs_converter --input_format tf_frozen_model --output_node_names 'MobilenetV2/Logits/output,labels' output_graph.pb ./js