Skip to content
/ MESH Public

Code for the paper "Unlocking Slot Attention by Changing Optimal Transport Costs"

License

Notifications You must be signed in to change notification settings

davzha/MESH

Repository files navigation

SA-MESH

[paper]

This repo implements the experiments in Unlocking Slot Attention by Changing Optimal Transport Costs.

To run the experiments, first prepare the data with the instructions provided in data_scripts/README.MD and install the necessary packages.

Experiments

mdsprites:

python run_object_discovery.py data=mdsprites model=image model/slot_attention=mesh

ClevrTex:

python run_object_discovery.py data=clevrtex64 model=image model/slot_attention=mesh

CLEVRER-S:

python run_object_discovery.py data=clevrer_s model=video model/slot_attention=mesh

CLEVRER-L:

python run_object_discovery.py data=clevrer_l model=video model/slot_attention=mesh

About

Code for the paper "Unlocking Slot Attention by Changing Optimal Transport Costs"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages