ClaudiaAlvarezAparicio / petra

GNU Lesser General Public License v3.0
22 stars 3 forks source link

PeTra (People Tracking)

PeTra is a tool which allows detecting and tracking people. The system is based on a CNN that uses an occupancy map constructed from the readings of a LIDAR sensor.

Getting Started

These instructions will get you a copy of the project up and running on your local machine for development and testing purposes. See deployment for notes on how to deploy the project on a live system.

Prerequisites

$ cd ~/catkin_ws/
$ catkin_make  
$ cd ~/catkin_ws/src/
$ git clone https://github.com/tradr-project/tensorflow_ros_cpp.git
$ cd ..
$ catkin_make

PeTra Installation

 $ cd ~/catkin_ws/src/  
 $ git clone https://github.com/ClaudiaAlvarezAparicio/petra.git
 $ cd ..  
 $ catkin_make  

Configuration

Edit the different parameters of petra/config/parameters.yaml

Execution

Execution with a rosbag:

roslaunch petra petra_rosbag.launch rosbag_file:=absolute_path_to_bag_file

Execution in real time:

roslaunch petra petra.launch

Docker Image

To test PeTra we have create tow docker images, steps to test it:

In the browser: http://localhost:6901/
Password: petra

Execute PeTra in docker

1.- Open terminal:

$ roscore  

2.- Open terminal:

$ rosrun rviz rviz  

3.- Open terminal

$ roslaunch petra petra_rosbag.launch rosbag_file:=/home/student/rosbags/kitchen1.bag  

Stop and Remove Docker Container

$ docker stop petra
$ docker rm petra 

Hardware Specifications

The tool has been test with a Hokuyo URG-04LX-UG01 (https://www.hokuyo-aut.jp/search/single.php?serial=166).

With another LIDAR it can work but is better train a new neural network model.

Data Labelig for Training

To train a new neural network model is necessary to get the LIDAR data and label it. The data_labeling folder has a README to label data with a KIO RTLS system or with the own PeTra.

Network Model Training

Once the data to train the network is ready, its time to train the model. The neural_network folder contains a README with the necessary steps to train the model.