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

[find_people] Detector replacement (SSD Keras -> PyTorch Faster R-CNN) #245

Closed alex-mitrevski closed 3 years ago

alex-mitrevski commented 3 years ago

Summary of changes

This PR replaces the SSD Keras model used in the find_people action with the PyTorch-based Faster R-CNN ResNet-50 FPN model. This is part of an effort to replace all Tensorflow-based models in our code base with PyTorch-based models (which already started with https://github.com/b-it-bots/mas_domestic_robotics/pull/207).

Related PRs:

Tests to verify the changes

I performed tests with Lucy for verifying the changes.

cc @AhmedFaisal95