b-it-bots / mas_domestic_robotics

Robot-independent ROS packages for domestic applications
GNU General Public License v3.0
28 stars 45 forks source link

[perceive_plane_action] Plane detector using a Torch-based object detector #247

Closed alex-mitrevski closed 3 years ago

alex-mitrevski commented 3 years ago

Summary of changes

This PR modifies the launch file of the perceive_plane action, which now loads a plane detector that uses a PyTorch-based detector (as in https://github.com/b-it-bots/mas_domestic_robotics/pull/245, Faster R-CNN ResNet-50 FPN model is used). This thus means that the repository does not depend on ssd_keras_ros anymore.

A config file for the detector is included under mdr_perceive_plane_action/config.

The PR also prevents the action from creating a recognition proxy unless this feature is enabled through the appropriate launch file parameter; otherwise, the proxy was created without being used internally.

Tests to verify the changes

The working of the modified component has been verified with a bag file that was recently collected from the HSR (here, the robot observes small objects on the floor).