利用神经网络进行超声波束合成:输入RF信号,通过神经网络进行波束合成
一、在 Colab 中使用 TensorFlow
需要使用 GPU,则点击菜单 “代码执行程序 - 更改运行时类型”,在 “硬件加速器” 一项中选择 “GPU”,如下图所示:
在 Colab 中,可以使用 !pip install 来安装 Colab 中尚未安装的 Python 库
!pip uninstall -y tensorflow
!pip install tensorflow-gpu==1.14.0
检查是否安装成功
二、训练波束合成神经网络
运行train.py
文件
!python train.py
输入数据:5000×64×64×32
, NWHC
测试数据:128×64×64×32
, NWHC
训练完成:
启动tensorboard
查看结果
三、使用PICMUS数据进行验证
! python predict.py
报错:
Traceback (most recent call last):
File "predict.py", line 68, in <module>
"../runs/20220107_0259/results_PICMUS.mat",
File "predict.py", line 49, in predict
model.load_weights(model_h5)
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/training.py", line 162, in load_weights
return super(Model, self).load_weights(filepath, by_name)
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/engine/network.py", line 1424, in load_weights
saving.load_weights_from_hdf5_group(f, self.layers)
File "/usr/local/lib/python3.7/dist-packages/tensorflow/python/keras/saving/hdf5_format.py", line 711, in load_weights_from_hdf5_group
original_keras_version = f.attrs['keras_version'].decode('utf8')
AttributeError: 'str' object has no attribute 'decode'
解决方案:
!pip install h5py==2.10.0
模型最终效果如下图