/Rgbd-GG-CNN

Rgbd Generative Grasping CNN (Rgbd-GG-CNN)

Primary LanguagePython

Rgbd-GG-CNN

A simple Rgbd Generative Grasping CNN (Rgbd-GG-CNN)

Here is the original paper:


Network Architecture V1

img

Network Architecture V2

improve:Use MobileNetV3 to extract Features img MobileNetV3(downsample 16 times),4 conv transpose layers(Upsample 16 times).

Network Architecture V3

improve:Add a fusion layer after feature extraction img

Train

train.py --dataset-path <Path>

Test

test.py --network epoch_04_iou_0.90 --dataset Oran --dataset-path <Path> --vis

Test result

img img img img

Dataset Example

img Download Link

Demo Video

Project1 Project2