Skip to content

Commit

Permalink
parameters are documented in the README
Browse files Browse the repository at this point in the history
  • Loading branch information
douglasrizzo committed Nov 1, 2018
1 parent fad6f9f commit 2a622ef
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 21 deletions.
17 changes: 14 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,24 @@ Check [dodo detector](https://github.com/douglasrizzo/dodo_detector)'s README fi

## Configuration

Edit `config/main_config.yaml` to select which type of detector you want as well as to point the package to the artifacts each detector expects.

- `global_frame`: the frame or tf that all object tfs will be published in relation to, eg `map`. Leave blank to publish wrt. camera_link
- `tf_prefix`: a prefix for the object tfs which will be published by the package
- `detector_type`: either 'sift', 'rootsift' or 'ssd'
- `inference_graph`: path to TensorFlow Object Detection API frozen inference graph, the `.pb` file
- `label_map`: path to TensorFlow Object Detection API label map, the `.pbtxt`
- `ssd_confidence`: confidence level to report objects as detected by the neural network, between 0 and 1
- `sift_min_pts`: minimum number of points to consider an object as present in the scene
- `sift_database_path`: path to the database used by the keypoint object detector

## Functionality

The package works in two steps. First, it detects objects by using a video feed and then it uses a point cloud to publish TFs for each detected object.

Object detection is done via the [dodo detector](https://github.com/douglasrizzo/dodo_detector) package. In case the single-shot detector is to be used, point the `inference_graph` and `label_map` parameters to your corresponding files. These files are created when training an object detection neural network using [TensorFlow Object Detection API](https://github.com/tensorflow/models/tree/master/research/object_detection). Also, the `ssd_confidence` parameter can be changed to adjust the detection threshold of the network.

In case the keypoint-based detector is to be used (either SIFT or RootSIFT), you need to create a database directory. The procedure to do so is described [here](http://dodo-detector.douglasrizzo.com.br/#keypoint-based-detector).

Edit `config/main_config.yaml` to select which type of detector you want as well as to point the package to the artifacts each detector expects.
In case the keypoint-based detector is to be used (either SIFT or RootSIFT), you need to create a database directory. The procedure to do so is described [here](http://douglasrizzo.github.io/dodo_detector#keypoint-based-detector).

## Usage

Expand Down
19 changes: 3 additions & 16 deletions config/main_config.yaml
Original file line number Diff line number Diff line change
@@ -1,21 +1,8 @@
# ['sift', 'rootsift', 'ssd']
# global_frame: map
tf_prefix: dodo_detector_ros
detector_type: ssd
# path to TensorFlow Object Detection API frozen inference graph
inference_graph: ~/.dodo_detector_ros/mscoco_frozen_inference_graph.pb
# path to TensorFlow Object Detection API label map
label_map: ~/.dodo_detector_ros/mscoco_label_map.pbtxt

# confidence level to report objects as detected
ssd_confidence: 0.5

# minimum number of points to consider an object as present in the scene
sift_min_pts: 10

# path to the database used by the keypoint object detector
sift_database_path: ~/.dodo_detector_ros/sift_database

# the frame or tf that all object tfs will be published in relation to, leave blank to publish wrt. camera_link
# global_frame: map

# a prefix for the object tfs which will be published by the package
tf_prefix: dodo_detector_ros
sift_database_path: ~/.dodo_detector_ros/sift_database
4 changes: 2 additions & 2 deletions package.xml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
<?xml version="1.0"?>
<package>
<name>dodo_detector_ros</name>
<version>0.2.0</version>
<version>0.2.1</version>
<description>A package for object detection and localization using computer vision and IR sensors.</description>

<!-- One maintainer tag required, multiple allowed, one person per tag -->
Expand All @@ -17,7 +17,7 @@
<!-- Url tags are optional, but multiple are allowed, one per tag -->
<!-- Optional attribute type can be: website, bugtracker, or repository -->
<!-- Example: -->
<!-- <url type="website">http://dodo-detector-ros.douglasrizzo.com.br/</url> -->
<!-- <url type="website">http://github.com/douglasrizzo/dodo_detector_ros</url> -->


<!-- Author tags are optional, multiple are allowed, one per tag -->
Expand Down

0 comments on commit 2a622ef

Please sign in to comment.