Skip to content

Commit

Permalink
set all config to correct default
Browse files Browse the repository at this point in the history
  • Loading branch information
gustavz committed May 24, 2018
1 parent 99ec7e6 commit 6575dd7
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 6 deletions.
13 changes: 10 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ Realtime Object Detection based on Tensorflow's [Object Detection API](https://g

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

## About the Project
The Idea was to create a scaleable realtime-capable object detection pipeline that runs on various systems. <br />
Plug and play, ready to use without deep previous knowledge.<br /> <br />
Expand All @@ -18,6 +19,7 @@ Results in up to x10 performance increase depending on the running system <br />
- run statistic tests on sets of images and get statistical information like mean and median fps, std dev and much more
- inspect, summarize, quantize, transform and benchmark models with the provided `scripts/`


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


## Tools:
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 />
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 />
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`.
To make use of the tools provided inside `scripts/` follow this guide: <br />
- first change all paths and variables inside `config_tools.sh` to your needs / according to your system
- 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 />
- 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`.


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


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


## Related Work:
- [test_models](https://github.com/GustavZ/test_models): A repo for models i am currently working on for benchmark tests
- [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 />
Expand Down
6 changes: 3 additions & 3 deletions scripts/config_tools.sh
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@

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

### MODEL TRANSFORMATION CONFIG ###
### CHANGE THIS ACCORDING TO YOUR MODEL ###
export SHAPE='1,400,400,3'
export SHAPE='1,300,300,3'
export STD_VALUE=127.5
export MEAN_VALUE=127.5
export INPUT_TYPE='uint8'
export INPUTS='image_tensor'
export OUTPUTS='num_detections,detection_boxes,detection_scores,detection_classes,detection_masks'
export OUTPUTS='num_detections,detection_boxes,detection_scores,detection_classes'
export TRANSFORMS='
add_default_attributes
strip_unused_nodes(type=float, shape=\"1,400,400,3\")
Expand Down

0 comments on commit 6575dd7

Please sign in to comment.