Skip to content

Commit 2fd7d45

Browse files
authored
link to tf wheel files added
1 parent 23708c8 commit 2fd7d45

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

README.md

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
# realtime_object_detection
22
Realtime object detection based on [Tensorflow's Object Detection API](https://github.com/tensorflow/models/tree/master/research/object_detection) with an extreme Focus on Performance. <br />
33
Specialized for `ssd_mobilenet` models.
4-
<br />
5-
<br />
4+
65
## About the Project
76
The Idea was to create a realtime capable object detection pipeline on various machines. <br />
87
Plug and play, ready to use without deep previous knowledge.<br /> <br />
@@ -19,7 +18,6 @@ The following work has been done based on the original API:
1918
Works only for `ssd_mobilenet` models but results in significant performance increase.
2019
- Added mutlithreading for the split sessions (thanks to [naisy](https://github.com/naisy))
2120
- **Results: Overall up to x10 Performance Increase** depending on the config and the running system
22-
<br />
2321

2422
## Getting Started:
2523
- create a copy of `config.sample.yml` called `config.yml`
@@ -28,7 +26,6 @@ Works only for `ssd_mobilenet` models but results in significant performance inc
2826
- run `image_detection.py` for single test image detection
2927
- run `object_detection.py` for realtime object detection
3028
- Enjoy!
31-
<br />
3229

3330
## Under Development (help appreciated):
3431
- **KCF Tracking**: run `./build_kcf.sh` inside directory, set `use_tracker` to `True` inside config, run `object_detection_kcf_test.py` (currently only works more or less stable without `split_model`)
@@ -38,14 +35,14 @@ Works only for `ssd_mobilenet` models but results in significant performance inc
3835
## My Setup:
3936
- Ubuntu 16.04
4037
- Python 2.7
41-
- Tensorflow 1.4
38+
- Tensorflow 1.4
39+
([this repo](https://github.com/peterlee0127/tensorflow-nvJetson) provides pre-build tf wheel files for jetson tx2)
4240
- OpenCV 3.3.1
43-
<br />
41+
> Note: This project currently does not run with tensorflow v1.7.0
4442
4543
## Current max Performance on `ssd_mobilenet` (with|without visualization):
4644
- Dell XPS 15 with i7 @ 2.80GHZ x8 and GeForce GTX 1050 4GB: **78fps | 105fps**
4745
- Nvidia Jetson Tx2 with Tegra 8GB: **30fps | 33 fps**
48-
<br />
4946

5047
## To Do:
5148
If you like the project, got improvement or constructive critisism, please feel free to open an Issue. <br />
@@ -54,7 +51,6 @@ Future implementation plans are: <br />
5451
- [ ] Add KCF Tracking to improve fps especially on the jetson
5552
- [ ] Mask-SSD: Modify SSD to be able to predict a segmentation mask in parallel to the bounding box
5653
- [ ] Split Model and Threading for R-CNN Models
57-
<br />
5854

5955
## Related Work:
6056
- [test_models](https://github.com/GustavZ/test_models): A repo for models i am currently working on for benchmark tests

0 commit comments

Comments
 (0)