From da29d12b5c5de2ac47f7d0970b35c6d63af583ef Mon Sep 17 00:00:00 2001 From: Aaron Bauman Date: Wed, 21 Mar 2018 18:37:31 -0400 Subject: [PATCH 1/2] replaced 'models-master' with 'models' project-wide, for consistency --- README.md | 6 +++--- analyzeimages.py | 2 +- generate_tfrecord.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 1a0ed84..1ad3093 100644 --- a/README.md +++ b/README.md @@ -20,7 +20,7 @@ The training data folder has a series of images and training annotations. Create a `test.record` file and a `train.record` file: ``` -./generate_tfrecord --folder=path/to/data_dir --train_ratio=.70 +python generate_tfrecord.py -folder=path/to/data_dir -train_ratio=.70 ``` Download [COCO-pretrained Faster R-CNN with Resnet-101 model](http://storage.googleapis.com/download.tensorflow.org/models/object_detection/faster_rcnn_resnet101_coco_11_06_2017.tar.gz) @@ -41,7 +41,7 @@ export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim 1) ``` -python models-master/research/object_detection/train.py --logtostderr +python models/research/object_detection/train.py --logtostderr --pipeline_config_path=./data/faster_rcnn_resnet101_cars.config --train_dir=./data/models/model/train ``` @@ -49,7 +49,7 @@ python models-master/research/object_detection/train.py --logtostderr 2) ``` -python models-master/research/object_detection/eval.py --logtostderr +python models/research/object_detection/eval.py --logtostderr --pipeline_config_path=./data/faster_rcnn_resnet101_cars.config --checkpoint_dir=./data/models/model/train --eval_dir=./data/models/model/eval diff --git a/analyzeimages.py b/analyzeimages.py index f3bbd5b..1d85b06 100644 --- a/analyzeimages.py +++ b/analyzeimages.py @@ -15,7 +15,7 @@ from matplotlib.ticker import FormatStrFormatter, FuncFormatter -sys.path.append('./models-master/research/') +sys.path.append('./models/research/') from object_detection.utils import label_map_util from object_detection.utils import visualization_utils as vis_util diff --git a/generate_tfrecord.py b/generate_tfrecord.py index c239b09..d400fb7 100644 --- a/generate_tfrecord.py +++ b/generate_tfrecord.py @@ -25,7 +25,7 @@ import tensorflow as tf import sys import xml.etree.ElementTree as eT -sys.path.append('./models-master/research/') +sys.path.append('./models/research/') from object_detection.utils import dataset_util From c225a195a0eb46f813f2fabf800b39906457580a Mon Sep 17 00:00:00 2001 From: Aaron Bauman Date: Wed, 21 Mar 2018 18:45:45 -0400 Subject: [PATCH 2/2] updated gitignore to exclude training data, cocoapi, and tensorflow --- .gitignore | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/.gitignore b/.gitignore index 0e97372..89c0ef0 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,10 @@ .DS_Store data/.DS_Store +.idea +cocoapi +data/models +data/test.record +data/train.record +models-master +models +tensorflow