This repository contains the code for the paper Tacchi: A Pluggable and Fast Gel Deformation Simulator for Optical Tactile Sensors. The code can be used for connection simulation with MuJoCo and Gazebo, Tacchi image generation, image quality comparison, and Sim2Real tasks. All the datasets generated by this repository can be found in Tacchi_dataset. The presentation video is available in Tacchi_video.
This package generates object files for Tacchi. To generate objects with different particles,
python stl2ply.py
python ply2pcd.py --particle [particle number]
python pcd2npy.py
This package connects MuJoCo and Gazebo with Tacchi for simulation. The objects generated by 0_stl2npy should be put in this package.
This package generates images with Tacchi. The objects generated by 0_stl2npy should be put in this package.
python cmd.py --particle [particle number]
This package is from the repository Gelsight Simulation. This package can align images and evaluate simulation images. The images form 2_Tacchi_image_generation should be put in this package. To align images,
python -m experiments.preparation.align_per_object
python -m experiments.preparation.generate_sim
To evaluate simulation images,
python -m experiments.preparation.loss_eval
This package applied depth map smoothing method applied in [18] to generate tactile images. To generate images,
python data_loader_AA21.py
This package applied ResNet18 to classifier images and evaluate images with Sim2Real tasks. The images form real experiments, Gelsight Simulation, and 2_Tacchi_image_generation should be put in this package. To train a ResNet,
python main.py --mode train --img_kind [image kind] --model_name [model name]
To test a classifier with real images,
python main.py --mode test --img_kind real --model_name [model name]
Thank Daniel Fernandes Gomes for Gelsight rendering method, alignment method and image evaluation code.