利用神经网络进行超声波束合成(Colab Tensorflow)

利用神经网络进行超声波束合成：输入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

模型最终效果如下图