This is repository for slip prediction of vision-based tactile sensor FingerVision.
git clone this repo down to a directory, then download dataset and unzip it along side with this repository folder.
- First check if the dataloader in convLSTM_dataset.py can run correctly by:
python convLSTM_dataset.py
This script defines a customized dataset for dataloader of pytorch.
- IF previous step passes, you can start training the frame prediction network by running convLSTM_frame_pred.py and should be able to see the drop of train/test loss versus epochs, also by:
python convLSTM_frame_pred.py
Meanwhile, run the following script in the shell and you should see the change of accuracy (finally it should hit over 95% of accuracy). This network is in charge of slip detection with a given frames of determined time steps (default T=10).
python convLSTM_slip_detection_1layer.py
- Finaly, run the slip prediction accuracy test by:
python convLSTM_pred_slip.py
and it would give out accuracy over 90% that depends on how many frames ahead it is iterating. This is a stacking of previous trained slip detection network and frame prediction network that features future slip prediction capability with high accuracy.
further test of generalization ability to different contact object with variant geometry, stiffness, and size, etc.