osrt/data/__init__.py

osrt/data/core.py

osrt/data/obsurf.py

osrt/data/ycb-video.py

osrt/model.py

outputs/visualisation_10.png

outputs/visualisation_2.png

outputs/visualisation_3.png

outputs/visualisation_4.png

outputs/visualisation_5.png

outputs/visualisation_6.png

outputs/visualisation_7.png

quick_test.py

runs/clevr/slot_att/config.yaml

runs/clevr/slot_att/config_tsa.yaml

runs/clevr3d/slot_att/config.yaml

runs/clevr3d/slot_att/config_tsa.yaml

train_sa.py

visualize_sa.py