Skip to content

Synthetic data set builder and object position detector based on Mask-RCNN

Notifications You must be signed in to change notification settings

chenyi0916/COBOT

Repository files navigation

The project aims to create an advanced collaborative assembly system to improve low-volume production lines. The main goals were to make processes flexible to adapt to customer demands quickly, ensure safe collaboration between humans and robots, and boost overall performance. This project involved various tasks, such as developing bin picking and assembling tools using standard equipment, designing workspaces where humans and robots could work together efficiently, and creating intelligent schedules for multiple robots in low-volume production. I collaborated on the Picking and Assembling part of the project. My assigned task involves detecting specific tagboard parts, including pose estimation and orientation detection, and integrating a predictor into the ROS pipeline for picking and assembling.

I used BlenderPoc2 to generate synthetic datasets which are devided into training set and validation set. Here are some examples of generated results:

1)various background:

image

2)workstation:

iamge

Afterwards, I trained datasets using the Mask R-CNN algorithm based on Detectron2. The following are the predicted results.

1)real image1:

image

2)real image2:

iamge

3)real image3:

image

The training loss is shown here:

image

It can be seen that the overall accuracy of the picked trained model is not bad. However, surrounding interfering objects, such as bolts, are also mistakenly detected as target objects. We believe this is because the distribution of feature data in the training set cannot be sufficiently similar to the ground truth.

About

Synthetic data set builder and object position detector based on Mask-RCNN

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages