欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python脚本运行gprMax3.0批量仿真GPR数据

程序员文章站 2022-03-28 22:32:33
python脚本运行gprMax3.0批量仿真GPR B-scan图像1.引言2.Python脚本3.可能出现的报错4.数据展示1.引言探地雷达(GPR)结合深度学习通常需要大量的训练数据集,对于GPR仿真数据集的获取,我们一般通过gprMax生成,而gprMax3.0仿真数据时需要通过cmd命令提示符窗口人工一条一条地输入指令(通过cmd命令生成GPR B-scan图像:可以参考我的上一篇博客.),对于批量生成GPR数据非常不方便。因此,有必要写一些Python脚本,实现自动化批量生成GPR数据集。...

python脚本运行gprMax3.0批量仿真GPR B-scan图像

1.引言

探地雷达(GPR)结合深度学习通常需要大量的训练数据集,对于GPR仿真数据集的获取,我们一般通过gprMax生成,而gprMax3.0仿真数据时需要通过cmd命令提示符窗口人工一条一条地输入指令(通过cmd命令生成GPR B-scan图像:可以参考我的上一篇博客.),对于批量生成GPR数据非常不方便。因此,有必要写一些Python脚本,实现自动化批量生成GPR数据集。

2.Python脚本

该Python脚本可以一次性批量生成指定文件夹中多个GPR输入文件(.in文件或者.txt文件)的输出文件(.out文件),并且可以把输出文件(.out文件)保存为数据格式(.txt文件)和图像格式(.png文件)。运行该脚本之前需要安装好gprMax3.0(可以参考我的上一篇博客

import sys
sys.path.append('D:/my_gprmax/gprMax') #把gprMax安装路径添加至系统,使import可以找到gprMax模块
import os
from gprMax.gprMax import api
import numpy as np
from tools.plot_Bscan import get_output_data, mpl_plot
from tools.outputfiles_merge import merge_files
import matplotlib.pyplot as plt


num_scan = 120 #正演仿真次数(A扫描次数)->B扫描
geo_only = False #是否只生成模型图

path = os.getcwd()#获得当前脚本所在的文件路径
#print(path)
root = path+'/in_data'   #指定输入文件的路径
files = os.listdir(root)   #得到路径下的文件夹名或者文件名,形成列表

for file in files:
    if file.endswith('.txt'):  #找到.txt文件
        # print(file)
        filename = root+'/'+file  #得到文件名的绝对路径
        fi = filename[0:-4]        #去掉文件名后的.txt后缀,注意保留了前面的路径
        api(filename, n=num_scan, geometry_only=geo_only) #geometry_only:仅几何图形
        merge_files(fi, removefiles=True)

        """B扫描绘图"""
        filename_b = fi+'_merged.out'
        rxnumber = 1
        rxcomponent = 'Ez'
        #获取回波数据
        outputdata, dt = get_output_data(filename_b, rxnumber, rxcomponent)
        #保存回波数据
        fi2 = fi.split('/')[-1]  #得到文件名,去掉了前面的路径
        out_path = path+'/out_data'
        if (os.path.exists(out_path)):
            pass
        else:
            os.mkdir(out_path)
        np.savetxt(out_path+'/'+fi2+'.txt',outputdata,delimiter=' ')#未去除直达波
        
        #绘图
        plt.imshow(outputdata, extent=[0, outputdata.shape[1], outputdata.shape[0], 0], interpolation='nearest', aspect='auto', cmap='gray',
                   vmin=-np.amax(np.abs(outputdata)), vmax=np.amax(np.abs(outputdata)))
        img_path = path+'/img_data'
        if (os.path.exists(img_path)):
            pass
        else:
            os.mkdir(img_path)
        plt.savefig(img_path+'/'+fi2+'.png',dpi=300)#保存图片
        # plt.show()

3.可能出现的报错

运行之后,如果出现报错:“no module named terminaltables",则打开cmd命令提示符窗口,使用“pip install terminaltables ”指令,安装好这个模块。若运行之后还出现报错:“no module named gprMax.fields_updates_ext",则gprMax3.0安装的时候可能出现了错误,我的解决办法是,下载:这个gprMax文件 (提取码:3ss8 ),并将这个文件替换gprMax文件夹下的gprMax模块。
python脚本运行gprMax3.0批量仿真GPR数据

4.数据展示

python脚本运行gprMax3.0批量仿真GPR数据
实验批量生成了3个输入文件的输出文件,结果如下:
python脚本运行gprMax3.0批量仿真GPR数据
python脚本运行gprMax3.0批量仿真GPR数据
python脚本运行gprMax3.0批量仿真GPR数据

本文地址:https://blog.csdn.net/weixin_43682976/article/details/107376980