Skip to content

Commit 6575dd7

Browse files
committed
set all config to correct default
1 parent 99ec7e6 commit 6575dd7

File tree

2 files changed

+13
-6
lines changed

2 files changed

+13
-6
lines changed

README.md

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ Realtime Object Detection based on Tensorflow's [Object Detection API](https://g
55

66
> Release Note: use Master or v2.0 to be additionally able to run and test Mask-Detection Models, KCF-Tracking and DeepLab Models (*merge of this [project](https://github.com/GustavZ/realtime_segmenation)*)
77
8+
89
## About the Project
910
The Idea was to create a scaleable realtime-capable object detection pipeline that runs on various systems. <br />
1011
Plug and play, ready to use without deep previous knowledge.<br /> <br />
@@ -18,6 +19,7 @@ Results in up to x10 performance increase depending on the running system <br />
1819
- run statistic tests on sets of images and get statistical information like mean and median fps, std dev and much more
1920
- inspect, summarize, quantize, transform and benchmark models with the provided `scripts/`
2021

22+
2123
## Inference:
2224
- create a copy of `config.sample.yml` called `config.yml`
2325
- optional: Change Parameters in `config.yml` to load other models or to modify configurations.<br />
@@ -30,10 +32,13 @@ or if you want to switch off the speed hack set `split_model` to `False`, <br />
3032
(put them as `.jpg` into `test_images/`)
3133
- Enjoy!
3234

35+
3336
## Tools:
34-
To make use of the tools provided inside `scripts/` first change all paths and params inside `config_tools.sh` to your needs / according to your system <br />
35-
When running **the first time** run `source config_tools.sh` and in the same terminal **run only once** `source build_tools.sh` to build the tools. this will take a while. <br />
36-
For all following uses **always from the same terminal** `source build_tools.sh`(due to the exported variables) and after that you are able to run the wanted scripts with `source script.sh`.
37+
To make use of the tools provided inside `scripts/` follow this guide: <br />
38+
- first change all paths and variables inside `config_tools.sh` to your needs / according to your system
39+
- When using **the first time** run: `source config_tools.sh` and in the same terminal **run only once** `source build_tools.sh` to build the tools. this will take a while. <br />
40+
- For all following uses first run: `source build_tools.sh`(due to the exported variables) and after that you are able to run the wanted scripts **always from the same terminal** with `source script.sh`.
41+
3742

3843
## Setup:
3944
Use the following setup for best and verified performance
@@ -48,6 +53,7 @@ Use the following setup for best and verified performance
4853
- Dell XPS 15 with i7 @ 2.80GHZ x8 and GeForce GTX 1050 4GB: **78fps | 105fps**
4954
- Nvidia Jetson Tx2 with Tegra 8GB: **30fps | 33 fps**
5055

56+
5157
## To Do:
5258
If you like the project, got improvement or constructive critisism, please feel free to open an Issue. <br />
5359
I am always happy to get feedback or help to be able to further improve the project. <br />
@@ -57,6 +63,7 @@ Future implementation plans are: <br />
5763
- [ ] Train a `mask_rcnn Model` with Mobilenet V1/V2 as backbone and deploy it on the Jetson
5864
- [ ] Split Model and Threading for R-CNN Models
5965

66+
6067
## Related Work:
6168
- [test_models](https://github.com/GustavZ/test_models): A repo for models i am currently working on for benchmark tests
6269
- [deeptraining_hands](https://github.com/GustavZ/deeptraining_hands): A repo for setting up the [ego](http://vision.soic.indiana.edu/projects/egohands/)- and [oxford](http://www.robots.ox.ac.uk/~vgg/data/hands/) hands-datasets.<br />

scripts/config_tools.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
### MODEL AND SYSTEM CONFIG ###
1111
### CHANGE THIS ACCORDING TO YOUR SYSTEM ###
12-
export MODEL_NAME="mask_rcnn_mobilenet_v1_400_coco_117k"
12+
export MODEL_NAME="ssd_mobilenet_v11_coco"
1313
export TF_PATH="/home/gustav/workspace/tensorflow/tensorflow"
1414
export ROOT_PATH="/home/gustav/workspace/realtime_object_detection"
1515
### DO NOT CHANGE THIS ###
@@ -20,12 +20,12 @@ export TFLITE_GRAPH="${MODEL_PATH}/frozen_inference_graph.tflite"
2020

2121
### MODEL TRANSFORMATION CONFIG ###
2222
### CHANGE THIS ACCORDING TO YOUR MODEL ###
23-
export SHAPE='1,400,400,3'
23+
export SHAPE='1,300,300,3'
2424
export STD_VALUE=127.5
2525
export MEAN_VALUE=127.5
2626
export INPUT_TYPE='uint8'
2727
export INPUTS='image_tensor'
28-
export OUTPUTS='num_detections,detection_boxes,detection_scores,detection_classes,detection_masks'
28+
export OUTPUTS='num_detections,detection_boxes,detection_scores,detection_classes'
2929
export TRANSFORMS='
3030
add_default_attributes
3131
strip_unused_nodes(type=float, shape=\"1,400,400,3\")

0 commit comments

Comments
 (0)