MCPcopy Create free account

hub / github.com/byaman14/SSDU / functions

Functions40 in github.com/byaman14/SSDU

↓ 5 callersFunctionconv_layer
Parameters ---------- x : input data conv_filter : weights of the filter is_relu : applies ReLU activation function is_scali
models/networks.py:5
↓ 2 callersFunctiontf_fftshift_flip2D
Parameters ---------- input_data : ncoil x nrow x ncol axes : The default is 1. ------
tf_utils.py:75
↓ 2 callersFunctiontf_ifftshift_flip2D
Parameters ---------- input_data : ncoil x nrow x ncol axes : The default is 1. ------
tf_utils.py:104
↓ 1 callersMethodEhE_Op
Performs (E^h*E+ mu*I) x
data_consistency.py:24
↓ 1 callersMethodGaussian_selection
(self, input_data, input_mask, std_scale=4, num_iter=1)
masks/ssdu_masks.py:33
↓ 1 callersMethodSSDU_kspace
Transforms unrolled network output to k-space and selects only loss mask locations(\Lambda) for computing loss
data_consistency.py:39
↓ 1 callersMethodSupervised_kspace
Transforms unrolled network output to k-space
data_consistency.py:52
↓ 1 callersMethodUnrolled_SSDU
(self)
UnrollNet.py:45
↓ 1 callersFunctionconj_grad
Parameters ---------- input_data : contains tuple of reg output rhs = E^h*y + mu*z , sens_maps and mask rhs = nrow x ncol x 2 se
data_consistency.py:64
↓ 1 callersFunctionifft
Parameters ---------- ispace : image space of size nrow x ncol x ncoil. axes : The default is (0, 1). norm : The default is N
utils.py:132
↓ 1 callersFunctionnorm
Parameters ---------- tensor : It can be in image space or k-space. axes : The default is (0, 1, 2). keepdims : The default is T
utils.py:161
↓ 1 callersMethoduniform_selection
(self, input_data, input_mask, num_iter=1)
masks/ssdu_masks.py:62
FunctionResNet
Parameters ---------- input_data : nrow x ncol x 2. Regularizer Input nb_res_blocks : default is 15. conv_filters : dictionary
models/networks.py:29
FunctionSSDU_kspace_transform
This function transforms unrolled network output to k-space at only unseen locations in training (\Lambda locations)
data_consistency.py:126
FunctionSupervised_kspace_transform
This function transforms unrolled network output to k-space
data_consistency.py:145
Method__init__
(self, sens_maps, mask)
data_consistency.py:16
Method__init__
(self, input_x, sens_maps, trn_mask, loss_mask)
UnrollNet.py:38
Method__init__
(self, rho=0.4, small_acs_block=(4, 4))
masks/ssdu_masks.py:29
Functionbody
(i, rsold, x, r, p, mu)
data_consistency.py:89
Functioncg_map_func
(input_elems)
data_consistency.py:116
Functioncomplex2real
Parameters ---------- input_data : row x col dtype :The default is np.float32. Returns ------- output : row x col x 2
utils.py:242
Functiondc_block
DC block employs conjugate gradient for data consistency,
data_consistency.py:111
Functionfft
Parameters ---------- ispace : coil images of size nrow x ncol x ncoil. axes : The default is (0, 1). norm : The default is N
utils.py:103
Functionfind_center_ind
Parameters ---------- kspace : nrow x ncol x ncoil. axes : The default is (1, 2, 3). Returns ------- the center of the
utils.py:182
FunctiongetPSNR
Measures PSNR between the reference and the reconstructed images
utils.py:92
FunctiongetSSIM
Measures SSIM between the reference and the reconstructed images
utils.py:76
Functionget_parser
()
parser_ops.py:4
Functionget_test_directory
Parameters ---------- args : args.data_opt--dataset to be used in training&testing Note: users should set the directories prior to r
utils.py:38
Functionget_train_directory
Parameters ---------- args : args.data_opt--dataset to be used in training&testing Note: users should set the directories prior to r
utils.py:5
Functionindex_flatten2nd
Parameters ---------- ind : 1D vector containing chosen locations. shape : shape of the matrix/tensor for mapping ind. Returns
utils.py:200
Functionmu_param
Penalty parameter used in DC units, x = (E^h E + \mu I)^-1 (E^h y + \mu * z)
models/networks.py:69
Functionreal2complex
Parameters ---------- input_data : row x col x 2 Returns ------- output : row x col
utils.py:258
Functionsense1
Parameters ---------- input_kspace : nrow x ncol x ncoil sens_maps : nrow x ncol x ncoil axes : The default is (0,1). Retur
utils.py:220
Functionssdu_map_fn
(input_elems)
data_consistency.py:133
Functionsupervised_map_fn
(input_elems)
data_consistency.py:152
Functiontest_graph
This function creates a test graph for testing
tf_utils.py:11
Functiontf_complex2real
Parameters ---------- input_data : nrow x ncol. Returns ------- outputs concatenated real and imaginary parts as nrow x ncol
tf_utils.py:45
Functiontf_fftshift
Parameters ---------- input_x : ncoil x nrow x ncol axes : The default is 1.
tf_utils.py:133
Functiontf_ifftshift
Parameters ---------- input_x : ncoil x nrow x ncol axes : The default is 1.
tf_utils.py:145
Functiontf_real2complex
Parameters ---------- input_data : nrow x ncol x 2 Returns ------- merges concatenated channels and outputs complex image of
tf_utils.py:60