This repo contains the physical property learner and the physics-based dynamic predictor of the Compositional Physics Learner in ComPhy.
- Download training and validation object tracks of target videos and reference videos extracted from Mask-RCNN;
- Download testing object tracks of target videos and reference videos extracted from Mask-RCNN.
Train mass predictor
sh scripts/train_mass_property_prp.sh
Train charge predictor
sh scripts/train_charge_property_prp.sh
Evaluate on validation set of mass predictor
sh scripts/test_mass_property.sh
Evaluate on validation set of charge predictor
sh scripts/test_charge_property.sh
process output of physical property learner for dynamic predictor
sh scripts/post_processing_prp.sh
Train dynamic predictor
sh scripts/train_dynamic_predictor.sh
Test on dynamic predictor
sh scripts/test_dynamic_predictor.sh
Much code is borrowed from Thomas Kipf's NRI repo.