Dual_model
├── model
│ ├── simulation
│ │ ├── results # IPE results
│ │ └── IPE.py # Running intuitive physics engine
│ └── model_compare.py # Model comparision script
├── pouring
│ ├── generate_diverse.py # Data simulation script
│ ├── make_videos.py # Video making tools
│ ├── simulator.py # Pouring simulator
│ └── tool.py # Other tools
├── stimuli
│ ├── round1-1
│ ├── round1-2
│ └── round1-3
├── config_diverse.yml # Data simulation configs
├── data.csv # Human data
└── README.md
conda create -n dual python=3.9
pip install pymunk pygame
pip install opencv-python
pip install matplotlib pandas numpy
pip instal statsmodels
Download the pouring-marble dataset from this link. The human results can be found in data.csv.
python ./pouring/generate_diverse.py
We implement a multi-process IPE to perform simulation under noise.
python ./model/simulation/IPE.py
Compare models including Deterministic physics, IPE, Heuristic model, and our SHM.
python ./model/model_compare.py