Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,2 +1,10 @@
.DS_Store
data/.DS_Store
.idea
cocoapi
data/models
data/test.record
data/train.record
models-master
models
tensorflow
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -41,15 +41,15 @@ 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
```

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
Expand Down
2 changes: 1 addition & 1 deletion analyzeimages.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
2 changes: 1 addition & 1 deletion generate_tfrecord.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down