/Wavepinns

Primary LanguageJupyter Notebook

基于波动方程约束的神经网络地震正反演算法

概述

我们使用在PINNs的基础上进行修改使其能运用到波动方程的地震正反演上。

安装

本仓库需要安装tensorflow用于实现PINNs

我们推荐一个新的工作环境，如下所式：

conda create -n Wavepinns python=3.6 # Use Anaconda package manager
conda activate Wavepinns

然后安装以下python依赖包：

pip install --ignore-installed --upgrade [packageURL]# install tensorflow (get packageURL from https://www.tensorflow.org/install/pip, see tensorflow website for details)
pip install tensorflow==1.15
pip install matplotlib
pip install pyDOE

以及画图需要的texlive环境

WavePINNs工作流程

首先介绍各个文件的用途：

  • generate_data用于使用SEISMIC_CMPL生成有限差分数据，并用matlab整理数据格式;
  • models存放训练好的神经网络
  • Pinns_class.py构建解决波动方程约束的神经网络

我们需要按照顺序执行以下脚本来训练网络：

  1. Train训练一个神经网络
  2. Wavepinns.ipynb调用训练好的神经网络画图