Skip to content

davidweizhongliu/DT-CycleGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 

Repository files navigation

DEMO
===============

###########Requirements
The usage of python package for this project
numpy==1.21.6
pybullet==3.2.5
timm==0.4.12
torch==1.9.1
torchvision==0.10.1
tqdm==4.62.3
opencv_python_headless==4.5.5.64
Pillow==9.2.0
visdom==0.1.8.9

###########File tree
All the files are as listed
├─GAN                    // Train the DT-CycleGAN, CyclGAN, RetinaGAN
└─Robot-FTC       // Train the Model on simulation enviroment
    ├─checkpoints
    ├─Env
    ├─ftc_robot     // the robots model for pybullet
    │  ├─config
    │  ├─launch
    │  ├─meshes
    │  └─urdf
    ├─Mesh           // background mesh figures
    │  └─swirly
    ├─objects         // grisp target


###########Usage
- ./Robot-FTC files contains the codes and model files of robots trained on the simulation enviroments. More details see ./Robot-FTC/README.md
- ./GAN files contains the codes and data of our the DT-CycleGAN methods, also with RetinaGAN and CycleGAN. MOre details see ./GAN/README.md

###########Contact

Yuzhong Chen - chenyuzhong211@gmail.com



About

Code for DT-CycleGAN visual grasping model

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors