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 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