Skip to content

Instantly share code, notes, and snippets.

@Jwata
Last active October 28, 2017 17:59
Show Gist options
  • Select an option

  • Save Jwata/e9f03054f70cec71ecdddcfc84a495f2 to your computer and use it in GitHub Desktop.

Select an option

Save Jwata/e9f03054f70cec71ecdddcfc84a495f2 to your computer and use it in GitHub Desktop.
Tensorflow Object Detection API クイックスタート ref: http://qiita.com/watanabe0621/items/08cc6c873066258435d6
git clone https://github.com/Jwata/tensorflow-pet-detector-quickstart
cd tensorflow-pet-detector-quickstart
pushd data
wget http://www.robots.ox.ac.uk/~vgg/data/pets/data/images.tar.gz
wget http://www.robots.ox.ac.uk/~vgg/data/pets/data/annotations.tar.gz
tar -xvf images.tar.gz
tar -xvf annotations.tar.gz
popd
docker run -it -v `pwd`/data:/data jwata/tensorflow-object-detection \
python object_detection/create_pet_tf_record.py \
--label_map_path=/data/pet_label_map.pbtxt \
--data_dir=/data \
--output_dir=/data
ls data/pet_*.record
data/pet_train.record data/pet_val.record
pushd data
wget http://storage.googleapis.com/download.tensorflow.org/models/object_detection/faster_rcnn_resnet101_coco_11_06_2017.tar.gz
tar -xvf faster_rcnn_resnet101_coco_11_06_2017.tar.gz
cp faster_rcnn_resnet101_coco_11_06_2017/* .
popd
docker run -d -v `pwd`/data:/data --name pet_detector_train jwata/tensorflow-object-detection \
python object_detection/train.py \
--logtostderr \
--pipeline_config_path=/data/faster_rcnn_resnet101_pets.config \
--train_dir=/data/train
docker run -d -v `pwd`/data:/data --name pet_detector_eval jwata/tensorflow-object-detection \
python object_detection/eval.py \
--logtostderr \
--pipeline_config_path=/data/faster_rcnn_resnet101_pets.config \
--checkpoint_dir=/data/train \
--eval_dir=/data/eval
docker run -d -v `pwd`/data:/data -p 6006:6006 --name tensorboard \
jwata/tensorflow-object-detection \
tensorboard --logdir=/data
open http://localhost:6006
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment