hitcslj / DiffAirfoil

DiffAirfoil: Editable and Controllable Airfoil Generation with Diffusion

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DiffAirfoil: Editable and Controllable Airfoil Generation with Diffusion

Installation

conda create --name airfoil python=3.8
conda activate airfoil
pip install -r requirements.txt

Dataset

请将数据集存放在 data 文件夹下, 默认数据集为 data/airfoil/supercritical_airfoil/*.dat

在项目的根文件夹下:

# split train/val/test
python dataload/datasplit.py 

# generate parsec feature
python dataload/parsec_direct.py 

Usage

在项目的根文件夹下:

# train vae, map airfoil to latent space
python train_vae.py

# train dit condition on keypoint & parsec
python train_dit.py

Resources

I have developed a baseline for editing airfoils using VAE. You can find the code here.

I have implemented a script to calculate the aerodynamic performance of airfoils, The code is available here.

I have created a demo for editing airfoils using Gradio. You can access the demo here.

I have curated a collection of papers on airfoil design. You can find the resources here.

Acknowledgements

I have intensively borrow codes from the following repositories. Many thanks to the authors for sharing their codes.

Citations

About

DiffAirfoil: Editable and Controllable Airfoil Generation with Diffusion

License:MIT License


Languages

Language:Python 100.0%