Skip to content

Commit 6ba6397

Browse files
committed
updated Readme to match latest status
1 parent eda8669 commit 6ba6397

File tree

1 file changed

+30
-27
lines changed

1 file changed

+30
-27
lines changed

README.md

Lines changed: 30 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,21 @@
11
# tf_vision
22

3-
On 15th June 2017 Google released their (pretrained) Computer Vision models via the TensorFlow Object Detection API.
3+
On 15th June 2017 Google released their first (pretrained) Computer Vision models via the TensorFlow Object Detection API.
44
These models won the COCO detection challenge (http://cocodataset.org/) in 2016. COCO features a large-scale dataset with more than 200k labeled images, 80 object categories and 250k people with keypoints.
55

6-
This repository derives from the Jupyter notebook of TensorFlow's Object Detection API and includes an Ansible playbook to easily setup your system to run (pretrained) models, to do fancy things incl. infering on live video data. Applying it to Robotics is one of the major goals.
6+
This repository builds upon the Jupyter notebook of TensorFlow's Object Detection API, includes an Ansible playbook to automatically setup your system to run (pretrained) models, allowing for interesting use cases when it comes to infering on video data. Applying it to Robotics is one of the major goals, though.
77

8-
Find here some highly recommended links:
9-
* Google announcement: https://research.googleblog.com/2017/06/supercharge-your-computer-vision-models.html
8+
Firstly, find here some highly recommended links:
9+
* Google initial announcement: https://research.googleblog.com/2017/06/supercharge-your-computer-vision-models.html
1010
* COCO detection challenge (incl. dataset, results): http://cocodataset.org/
1111
* CVPR 2017 Paper describing the models in detail: https://arxiv.org/abs/1611.10012
1212
* TensorFlow research model and object_detection API: https://github.com/tensorflow/models/tree/master/research/object_detection
1313

14-
This repository assumes you have an Ubuntu 16.04 x86_64 running Python 3.(5.X).
14+
This repository assumes you have an Ubuntu 16.04 x86_64. You can choose between using either Python 2.7 or 3.5.
1515

16-
# GPU Acceleration
16+
# Requirements for GPU Acceleration
1717

18-
> Note: As of current TensorFlow version 1.4.0: `All our prebuilt binaries have been built with CUDA 8 and cuDNN 6. We anticipate releasing TensorFlow 1.5 with CUDA 9 and cuDNN 7.` However, building from source with 1.4.0 using CUDA 9 and cuDNN 7 should be fine and working.
18+
> Note: As of current TensorFlow version 1.4.0: `All our prebuilt binaries have been built with CUDA 8 and cuDNN 6. We anticipate releasing TensorFlow 1.5 with CUDA 9 and cuDNN 7.` However, building from source with 1.4.0 using CUDA 9 and cuDNN 7 is working well already.
1919
2020
For GPU-accelerated tensorflow, ensure you do or have done the following:
2121

@@ -38,21 +38,21 @@ For GPU-accelerated tensorflow, ensure you do or have done the following:
3838

3939
Then switch to respective latest stable version branch (System Settings -> Software & Updates -> Additional Drivers).
4040

41-
3. Install latest CUDA:
41+
3. Install CUDA:
4242

43-
Download the latest NVIDIA CUDA Toolkit at https://developer.nvidia.com/cuda-downloads (select `deb (local)` in the end), then follow the Installation Instructions below.
43+
Download the respective NVIDIA CUDA Toolkit at https://developer.nvidia.com/cuda-downloads (select `deb (local)` in the end), then follow the Installation Instructions below.
4444

4545
For further notes, refer to: http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
4646

4747
To uninstall previous versions please refer to: http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#handle-uninstallation
4848

4949
Note: Our Ansible provisioning will take care of setting PATH and LD_LIBRARY_PATH environment variables properly.
5050

51-
3. Install latest NVIDIA CUDA Deep Neural Network library (cuDNN)
51+
3. Install NVIDIA CUDA Deep Neural Network library (cuDNN)
5252

5353
First, you need to register for the NVIDIA Developer Program at https://developer.nvidia.com/accelerated-computing-developer
5454

55-
Then, download latest version at https://developer.nvidia.com/rdp/cudnn-download of the following Debian Files:
55+
Then, download respective version at https://developer.nvidia.com/rdp/cudnn-download of the following Debian Files (e.g.):
5656

5757
* cuDNN v7.0 Runtime Library for Ubuntu16.04 (Deb)
5858
* cuDNN v7.0 Developer Library for Ubuntu16.04 (Deb)
@@ -72,31 +72,34 @@ For GPU-accelerated tensorflow, ensure you do or have done the following:
7272

7373
1. Setup log folder for Ansible (replace `sjentzsch` with your local user to run Ansible from):
7474

75-
```
76-
sudo mkdir -p /var/log/ansible/ && sudo chown -R sjentzsch:adm /var/log/ansible/ && sudo chmod 2750 /var/log/ansible/
77-
```
75+
```
76+
sudo mkdir -p /var/log/ansible/ && sudo chown -R sjentzsch:adm /var/log/ansible/ && sudo chmod 2750 /var/log/ansible/
77+
```
7878

79-
2. Create a new and clean folder called e.g. `ml` somewhere, and inside, checkout `setup_common_lib` and this GitHub repository (here with SSH keys):
79+
2. Create a new and clean folder called e.g. `ml` somewhere, and inside, clone `setup_common_lib` and this GitHub repository (here with SSH keys):
8080

81-
```
82-
git clone ssh://git@scr.bsh-sdd.com:7999/civtm/setup_common_lib.git
83-
git clone git@github.com:sjentzsch/tf_vision.git
84-
```
81+
```
82+
git clone ssh://git@scr.bsh-sdd.com:7999/civtm/setup_common_lib.git
83+
git clone git@github.com:sjentzsch/tf_vision.git
84+
```
8585

86-
Note: Currently, the shared library `setup_common_lib.git` lives inside BSH SDD.
86+
Note: Currently, the repository `setup_common_lib` lives inside restricted BSH SDD.
8787

88-
3. Inside `ml/tf_vision/ansible/` setup your computer by running Ansible via:
88+
3. Inside `ml/tf_vision/ansible/` setup your machine by running Ansible:
8989

90-
```
91-
ansible-playbook -i hosts -v local.yml -K
92-
```
90+
```
91+
ansible-playbook -i hosts -v local.yml -K
92+
```
9393

94-
Ansible will immediately ask you for your sudo password, as packages etc. will need to be installed through root user.
94+
Ansible will immediately ask you for your sudo password, as packages etc. will need to be installed through root user.
9595

9696
# Run
9797

98-
Inside `ml/tf_vision/` run:
98+
Inside `ml/tf_vision/` simply run:
9999

100100
```
101-
python3 object_detection_tutorial.py
101+
# Python 2.7
102+
python obj_detect.py
103+
# Python 3.5
104+
python3 obj_detect.py
102105
```

0 commit comments

Comments
 (0)