Data

Tools

Indexes

Applications

Experiments

Open Source Science

https://github.com/ahenkes1/meshgraphnets_pytorch

PyTorch implementations of Learning Mesh-based Simulation With Graph Networks

https://github.com/ahenkes1/meshgraphnets_pytorch

Science Score: 10.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • ○
    CITATION.cff file
  • ○
    codemeta.json file
  • ○
    .zenodo.json file
  • ○
    DOI references
  • ✓
    Academic publication links
    Links to: arxiv.org
  • ○
    Academic email domains
  • ○
    Institutional organization owner
  • ○
    JOSS paper metadata
  • ○
    Scientific vocabulary similarity
    Low similarity (11.5%) to scientific vocabulary
Last synced: 9 months ago · JSON representation

Repository

PyTorch implementations of Learning Mesh-based Simulation With Graph Networks

Basic Info
  • Host: GitHub
  • Owner: ahenkes1
  • License: apache-2.0
  • Language: Python
  • Default Branch: master
  • Homepage:
  • Size: 75.4 MB
Statistics
  • Stars: 1
  • Watchers: 0
  • Forks: 1
  • Open Issues: 0
  • Releases: 0
Fork of echowve/meshGraphNets_pytorch
Created almost 3 years ago · Last pushed over 2 years ago
Metadata Files
Readme License

README.md

Learning Mesh-Based Simulation with Graph Networks

This repository contains PyTorch implementations of meshgraphnets for flow around circular cylinder problem on the basic of PyG (pytorch geometric).

The original paper can be found as following:

Pfaff T, Fortunato M, Sanchez-Gonzalez A, et al. Learning mesh-based simulation with graph networks[J]. International Conference on Learning Representations (ICLR), 2021.

Some code of this repository refer to Differentiable Physics-informed Graph Networks.

Authors

  • Jiang
  • Zhang
  • Chu
  • Qian
  • Li
  • Wang

Requirements

  • h5py==3.6.0
  • matplotlib==3.4.3
  • numpy==1.21.1
  • opencv_python==4.5.4.58
  • Pillow==9.1.0
  • torch==1.9.0+cu111
  • torch_geometric==2.0.4
  • torch_scatter==2.0.8
  • tqdm==4.62.3

bash pip install -r requirements.txt

Sample usage

  • Download cylinder_flow dataset using the script https://github.com/deepmind/deepmind-research/blob/master/meshgraphnets/download_dataset.sh.

  • Parse the downloaded dataset into .h5 file using the tool parse_tfrecord.py

  • Change the dataset_dir in train.py to your .h5 files.

  • train the model by run python train.py.

  • For test, run rollout.py, and the result pickle file will be saved at result folder, the you can run the render_results.py to generate result videos that can be saved at videos folder.

Demos

  • Here are some examples, trained on cylinder_flow dataset.

  • In addition, we use simulation software to generate new training data. The test results on our data are as following:

Contact me

:email: jianglx@whu.edu.cn

Owner

  • Name: Alexander Henkes
  • Login: ahenkes1
  • Kind: user

GitHub Events

Total
Last Year

Issues and Pull Requests

Last synced: 9 months ago

All Time
  • Total issues: 0
  • Total pull requests: 1
  • Average time to close issues: N/A
  • Average time to close pull requests: 19 days
  • Total issue authors: 0
  • Total pull request authors: 1
  • Average comments per issue: 0
  • Average comments per pull request: 0.0
  • Merged pull requests: 1
  • Bot issues: 0
  • Bot pull requests: 0
Past Year
  • Issues: 0
  • Pull requests: 0
  • Average time to close issues: N/A
  • Average time to close pull requests: N/A
  • Issue authors: 0
  • Pull request authors: 0
  • Average comments per issue: 0
  • Average comments per pull request: 0
  • Merged pull requests: 0
  • Bot issues: 0
  • Bot pull requests: 0
View more stats
Top Authors
Issue Authors
Pull Request Authors
  • mathmerizing (1)
Top Labels
Issue Labels
Pull Request Labels

Dependencies

requirements.txt pypi
  • Pillow ==9.1.0
  • h5py ==3.6.0
  • matplotlib ==3.4.3
  • numpy ==1.21.1
  • opencv_python ==4.5.4.58
  • torch ==1.9.0
  • torch_geometric ==2.0.4
  • torch_scatter ==2.0.8
  • tqdm ==4.62.3