Ning Guo, Xudong Han, Xiaobo Liu, Shuqiao Zhong, Zhiyuan Zhou, Jian Lin, Jiansheng Dai, Fang Wan*, Chaoyang Song*
- Download Soft Finger Dataset.
- Extract "dataset" folder to root directory of this git repository : /path/to/SoftFingerSvae
- Cnn regression model: SoftFinger_cnn.py
- Vae model: SoftFinger_Vae.py
- Svae model: SoftFinger_Svae.py
This code was developed with Python 3.8 on Ubuntu 18.04. Additional Python packages:
- pytorch
- pytorch_lightning
- torchmetrics
- numpy
- torchvision
- pandas
- skimage
- PILLOW