You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
4
4
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.
5
5
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.
7
7
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
* CVPR 2017 Paper describing the models in detail: https://arxiv.org/abs/1611.10012
12
12
* TensorFlow research model and object_detection API: https://github.com/tensorflow/models/tree/master/research/object_detection
13
13
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.
15
15
16
-
# GPU Acceleration
16
+
# Requirements for GPU Acceleration
17
17
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.
19
19
20
20
For GPU-accelerated tensorflow, ensure you do or have done the following:
21
21
@@ -38,21 +38,21 @@ For GPU-accelerated tensorflow, ensure you do or have done the following:
38
38
39
39
Then switch to respective latest stable version branch (System Settings -> Software & Updates -> Additional Drivers).
40
40
41
-
3. Install latest CUDA:
41
+
3. Install CUDA:
42
42
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.
44
44
45
45
For further notes, refer to: http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
46
46
47
47
To uninstall previous versions please refer to: http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#handle-uninstallation
48
48
49
49
Note: Our Ansible provisioning will take care of setting PATH and LD_LIBRARY_PATH environment variables properly.
50
50
51
-
3. Install latest NVIDIA CUDA Deep Neural Network library (cuDNN)
51
+
3. Install NVIDIA CUDA Deep Neural Network library (cuDNN)
52
52
53
53
First, you need to register for the NVIDIA Developer Program at https://developer.nvidia.com/accelerated-computing-developer
54
54
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.):
56
56
57
57
* cuDNN v7.0 Runtime Library for Ubuntu16.04 (Deb)
58
58
* 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:
72
72
73
73
1. Setup log folder for Ansible (replace `sjentzsch` with your local user to run Ansible from):
0 commit comments