DL for speed of sound inversion in ultrasound imaging
Collection
2 items
•
Updated
•
1
This repository contains the code and models for the following papers:
This repository contain the network code and models for the algorithms and results contained in the paper.
The code was tested under python 3.9. The anaconda environment is defined in environment.yml (setup environment with the command conda env create -f environment.yml
)
The dataset used is available on huggingface at https://huggingface.co/datasets/laughingrice/Ultrasound_planewave_sos_inversion
Variables in the files are [sample, layer, x/channel, y/sample]
order
alpha_coeff
-- Alpha coefficient used for simulations, full resolutionc0
-- Speed-of-sound used for simulations, full resolutiondata
-- Channel data (first 2048 samples, 64 active channels, first layer with flat plane wave, to
match existing physical hardware were used for the results in the paper)dx
-- spatial dx value of c0
and alpha_coef
f
-- temporal sampling frequency of channel data (40MHz)Model files appearing under the models
directory for results presented in the paper with teh matching
execution parameters are as follows:
tbme_sos.pt
-- network weights for the network presented in [1]python . --test_files data/supplamentary_sample.mat --test_fname tbme_sos.h5 --load_ver models/tbme_sos.pt --net_type tbme
embc_sos.pt
-- network weights for the network presented in [2] python . --test_files data/supplamentary_sample.mat --test_fname embc_sos.h5 --load_ver models/embc_sos.pt --net_type embc
tbme2_sos.pt
-- network weights for the network presented in [3]python . --test_files data/supplamentary_sample.mat --test_fname tbme2_sos.h5 --load_ver models/tbme2_sos.pt
tbme2_sos_rand_gain.pt
-- [3] trained to recover the speed-of-sound map with random gain profile and scalingpython . --test_files data/supplamentary_sample.mat --test_fname tbme2_sos_gain.h5 --load_ver models/tbme2_sos_rand_gain.pt
tbme2_attn.pt
-- [3] trained to recover the attenuation coefficientpython . --test_files data/supplamentary_sample.mat --test_fname tbme2_attn.h5 --load_ver models/tbme2_attn.pt --label_vars alpha_coeff
tbme2_sos_attn.pt
-- [3] trained to recover both the speed-of-sound map and attenuation coefficientpython . --test_files data/supplamentary_sample.mat --test_fname tbme2_sos_attn.h5 --load_ver models/tbme2_sos_attn.pt --label_vars c0 alpha_coeff
tbme2_phase_sos.pt
-- [3] trained to recover the speed-of-sound map using the IQ phase componentpython . --test_files data/supplamentary_sample.mat --test_fname tbme2_phase_sos.h5 --load_ver models/tbme2_phase_sos.pt --phase_inv 1