fredfyyang / Touch-and-Go

22 stars 4 forks source link

Hard/soft classification #6

Open AntonioStefani opened 6 months ago

AntonioStefani commented 6 months ago

Hi Fred! I was looking at your code for the "Visuo-tactile contrastive learning" task. From what I understand from the Touch&Go paper, I expected to perform classification in 3 aspects: material (20 classes), roughness (2 classes), hardness (2 classes). However in the dataset folder I find only the files related to the material and roughness classification tasks. Also, in dataset.py I see that only the material and roughness cases are considered, so I was wondering if I am missing something. Thank you very much in advance!