Training and prediction code is in fit_primitives.ipynb notebook.
For Chair-Table dataset:
- Download the dataset from https://drive.google.com/file/d/1YEb68jeCwm0okCA3YRXinYBZo324zxgO/view?usp=sharing
- Create a folder "datasets" and move the "table_chair.h5" file to datasets folder.