Skip to content

cgraber/cvpr_dNRI

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 

Code accompanying "Dynamic Neural Relational Inference"

This codebase accompanies the paper "Dynamic Neural Relational Inference" from CVPR 2020.

This code was written using the following packages:

  • PyTorch 1.2.0
  • numpy 1.16.4
  • transforms3d 0.3.1 (For Motion Capture data processing)
  • pandas (for InD data processing)

To run this code, you should pip install it in editable mode. This can be done using the following command:

pip install -e ./

Scripts train models can be found in the run_scripts directory.

Datasets:

Attribution: Some portions of this code are based on the code for the paper "Neural Relational Inference for Interacting Systems." This code can be found at https://github.com/ethanfetaya/NRI

If you use this code or this model in your work, please cite us:

@inproceedings{dNRI,
  title={Dynamic Neural Relational Inference},
  author={Graber, Colin and Schwing, Alexander},
  booktitle={The IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
  year={2020},
}

About

Code accompanying "Dynamic Neural Relational Inference" from CVPR 2020

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published