Skip to content

Autonomous Drive Turtlebot using BiSeNet segmentation network on RGBD data

Notifications You must be signed in to change notification settings

zju-vipa/Wali-turtlebot

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Wali-turtlebot

Wali turtlebot is a self-driving turtlebot, which uses Scene Segmentation on RGBD data to make Path Planning for turtlebot. This process is shown in the following picture.

auto_drive

1. Hardware

The hardware device we use:

  • Turtlebot2
  • HIKVISION wireless camera (stereo)
  • Microsoft Kinect v1
  • HiSilicon970 (arm)
  • others, little computer host, Intel RealSense R200

We use a kinect v1 to obtain RGB and Depth data now, in the future, we'll replace it with the stereo camera composed of two HIKVISION wireless cameras.

There are 2 reasons why we use stereo rather than kinect.

  • kinect can't obtain intense depth image.
  • kienct can't work well in outdoor environment.

The following picture shows the evolution process of Wali turtlebot.

ks

2. Technology

The core technology we use:

  • Bilateral Semantic Segmentation on RGBD data (BiSeNet-RGBD, 20fps on Nvidia Quadro P5000)
  • ROS robot nodes communication mechanism
  • Turtlebot motion control using rospy (forward, left, right, back, and smoothly speed up)
  • Depth-based direction choose if not use the neural network(choose the direction with the largest depth)

2.1 BiSeNet-RGBD

BiSeNet-RGBD architecture is shown below.

bisenet_rgbd

BiSeNet-RGBD is trained on Princeton SUN-RGBD dataset. Now it can predict 37 class, we'll annotate some specific classes in our practical scenario using labelme in the future.

SUNRGB_37_label_map.png

Test scenes:There are 10 scenes including 4 indoors and 6 outdoors, which are used to test model performance. The test results are list in part 3.

2.2 Wali turtlebot control system

wali_arc.png

By using this architecture, we've made some drive test in the real scenario.

The test videos are shown in part 3.

3. Test results

3.1 BiSeNet-RGBD

  • 4 indoor scenes

in1.png

in2.png

in3.png

in4.png

  • 6 outdoor scenes

out1.png

out2.png

out3.png

out4.png

out5.png

out6.png

3.2 Wali drive test video

未设置匀变速运动,速度突变卡顿

添加匀变速运动之后,卡顿问题解决

updating...

About

Autonomous Drive Turtlebot using BiSeNet segmentation network on RGBD data

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 77.2%
  • CMake 22.5%
  • Shell 0.3%