MCPcopy Create free account

hub / github.com/dek924/PerX2CT / functions

Functions506 in github.com/dek924/PerX2CT

⨍ Functions 506◇ Types & classes 129
↓ 20 callersMethodget_last_layer
(self)
x2ct_nerf/models/aegan.py:174
↓ 19 callersMethodwrite
(self, string, end='\n')
utils/logger/TestLogger.py:32
↓ 18 callersFunctionnonlinearity
(x)
x2ct_nerf/modules/diffusionmodules/model.py:30
↓ 15 callersMethod__init__
(self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks, attn_resolutions, dropout=0.0, re
x2ct_nerf/modules/diffusionmodules/model.py:284
↓ 15 callersMethodlog_loss
(self, loss_name, loss)
utils/logger/TestLogger.py:46
↓ 14 callersFunctionNormalize
(in_channels)
x2ct_nerf/modules/diffusionmodules/model.py:35
↓ 12 callersFunctioninstantiate_from_config
(config)
main.py:129
↓ 12 callersMethodprint_loss
(self, loss_dict)
x2ct_nerf/models/aegan.py:116
↓ 11 callersMethodeval
(self)
taming/modules/misc/coord.py:8
↓ 11 callersFunctionnonlinearity
(x)
taming/modules/diffusionmodules/model.py:29
↓ 10 callersMethod__init__
(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks, attn_resolutions, dropout=0.0, resam
taming/modules/diffusionmodules/model.py:196
↓ 9 callersFunctionNormalize
(in_channels)
taming/modules/diffusionmodules/model.py:34
↓ 9 callersMethoddecode
(self, feature)
x2ct_nerf/models/aegan.py:80
↓ 7 callersMethod__init__
(self, H, W)
taming/modules/transformer/permuter.py:63
↓ 7 callersFunctiondefine_eval_dict
(sub_type={})
main_test_zoom.py:254
↓ 7 callersMethodlog_images
(self, batch, **kwargs)
x2ct_nerf/models/base.py:241
↓ 6 callersFunctionload_scan_mhda
(path)
data_preprocessing/ctxray_utils.py:32
↓ 6 callersFunctiontensor_back_to_unMinMax
(input_image, min, max)
utils/metrics.py:16
↓ 5 callersMethod__init__
(self, vocab_size, block_size, n_layer=12, n_head=8, n_embd=256, embd_pdrop=0., resid_pdrop=0
taming/modules/transformer/mingpt.py:127
↓ 5 callersMethod_block
(in_channels, features, name)
x2ct_nerf/modules/image_encoder/resnet.py:222
↓ 5 callersMethodencode
(self, x)
x2ct_nerf/models/aegan.py:76
↓ 5 callersMethodget_input
(self, batch, k)
x2ct_nerf/models/aegan.py:108
↓ 5 callersFunctionnormalize_vecs
Normalize vector lengths.
x2ct_nerf/modules/volume_rendering.py:99
↓ 4 callersMethod__init__
(self, with_r=False)
x2ct_nerf/modules/discriminator/base.py:36
↓ 4 callersMethod__init__
(self, ddconfig, lossconfig, n_embed, embe
x2ct_nerf/models/aegan.py:20
↓ 4 callersMethod__init__
(self, num_tokens, codebook_dim, decay=0.99, eps=1e-5)
taming/modules/vqvae/quantize.py:334
↓ 4 callersFunctionadopt_weight
(weight, global_step, threshold=0, value=0.)
x2ct_nerf/modules/losses/vqperceptual.py:14
↓ 4 callersFunctionget_ckpt_path
(name, root, check=False)
taming/util.py:36
↓ 4 callersMethodget_input
(self, batch)
x2ct_nerf/models/base.py:175
↓ 4 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
x2ct_nerf/models/aegan.py:65
↓ 4 callersMethodprepare_data
(self)
main.py:166
↓ 4 callersMethodprint_loss
(self, loss_dict)
x2ct_nerf/models/base.py:75
↓ 4 callersMethodsetup
(self, stage=None)
main.py:170
↓ 4 callersMethodshared_step
(self, batch, split, optimizer_idx)
x2ct_nerf/models/aegan.py:411
↓ 3 callersFunctionPeak_Signal_to_Noise_Rate
:param arr1: Format-[NDHW], OriImage [0,1] :param arr2: Format-[NDHW], ComparedImage [0,1] :return: Format-None if size
utils/metrics.py:169
↓ 3 callersFunctionPeak_Signal_to_Noise_Rate_3D
:param arr1: Format-[NDHW], OriImage [0,1] :param arr2: Format-[NDHW], ComparedImage [0,1] :return: Format-None if size
utils/metrics.py:136
↓ 3 callersFunctionPeak_Signal_to_Noise_Rate_total
(tensor1, tensor2, size_average=True, PIXEL_MIN=0.0, PIXEL_MAX=1.0, use_real_max=False)
utils/metrics.py:20
↓ 3 callersFunctionStructural_Similarity_slice
:param arr1: B x C x H x W, OriImage [0,1] :param arr2: B x C x H x W, ComparedImage [0,1] :param channel_axis : which axis is channel?
utils/metrics.py:289
↓ 3 callersMethod__init__
(self, batch_frequency, max_images, clamp=True, increase_log_steps=True)
main.py:285
↓ 3 callersMethod__init__
(self, use_dropout=True)
x2ct_nerf/modules/losses/lpips.py:13
↓ 3 callersMethod__init__
(self, in_channels, out_channels)
x2ct_nerf/modules/image_encoder/resnet.py:190
↓ 3 callersMethod__init__
(self, num_features, logdet=False, affine=True, allow_reverse_init=False)
taming/modules/util.py:11
↓ 3 callersMethod__init__
(self, use_dropout=True)
taming/modules/losses/lpips.py:13
↓ 3 callersMethodencode
NS : number of source src_images : NS x B x C x H x W src_camposes : (NS x B x 2), 2 is pitch and yaw or None render_
x2ct_nerf/modules/inr/model.py:44
↓ 3 callersFunctionget_data_transform
(in_channels, input_img_size)
x2ct_nerf/modules/image_encoder/get_transform.py:10
↓ 3 callersMethodget_image
(self, image_path, data_type='ct')
x2ct_nerf/data/base.py:60
↓ 3 callersMethodget_input
(self, batch)
x2ct_nerf/models/zoom_aegan.py:52
↓ 3 callersMethodget_input
(self, batch, k)
x2ct_nerf/models/base.py:67
↓ 3 callersMethodget_input
(self, batch)
x2ct_nerf/models/aegan.py:250
↓ 3 callersMethodget_input
(self, batch)
x2ct_nerf/models/aegan.py:559
↓ 3 callersMethodget_input
(self, batch)
x2ct_nerf/models/aegan.py:725
↓ 3 callersMethodget_p0_zoom_size
(self, ct_size_xy)
x2ct_nerf/modules/INREncoderZoomAxisInAlign.py:82
↓ 3 callersMethodget_p0_zoom_size
(self, ct_size_xy)
x2ct_nerf/modules/INRGLEncoderZoomAxisInAlign.py:99
↓ 3 callersFunctionset_requires_grad
(model, value)
x2ct_nerf/modules/utils.py:1
↓ 2 callersFunctionPeak_Signal_to_Noise_Rate_2D
(tensor1, tensor2, size_average=True, PIXEL_MIN=0.0, PIXEL_MAX=1.0, use_real_max=False)
utils/metrics.py:32
↓ 2 callersFunctionStructural_Similarity
:param arr1: Format-[NDHW], OriImage [0,1] :param arr2: Format-[NDHW], ComparedImage [0,1] :return: Format-None if size
utils/metrics.py:218
↓ 2 callersMethod__init__
(self)
x2ct_nerf/modules/losses/vqperceptual.py:10
↓ 2 callersMethod__init__
(self, *args, **kwargs)
x2ct_nerf/data/LIDC.py:5
↓ 2 callersFunction_isArrayLike
(obj)
x2ct_nerf/preprocessing/X2CT_transform_3d.py:48
↓ 2 callersFunctionadopt_weight
(weight, global_step, threshold=0, value=0.)
x2ct_nerf/modules/losses/perceptual.py:14
↓ 2 callersFunctionadopt_weight
(weight, global_step, threshold=0, value=0.)
taming/modules/losses/vqperceptual.py:14
↓ 2 callersFunctioncalculate_metric_per_patient
(recon, gt, curr_axis, DEVICE, logger, ct_min_max, batch_idx, save_freq, dicts,
main_test.py:137
↓ 2 callersFunctioncreate_cam2world_matrix
Takes in the direction the camera is pointing and the camera origin and returns a cam2world matrix.
x2ct_nerf/modules/volume_rendering.py:286
↓ 2 callersMethoddecode
(self, feature, skipfeatures)
x2ct_nerf/models/aegan.py:575
↓ 2 callersMethodget_last_layer
(self, key)
x2ct_nerf/models/aegan.py:389
↓ 2 callersMethodget_model
(self, module_name)
x2ct_nerf/modules/inr/model_global.py:29
↓ 2 callersMethodget_model
(self, module_name)
x2ct_nerf/modules/inr/model.py:31
↓ 2 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
x2ct_nerf/modules/diffusionmodules/model.py:9
↓ 2 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
taming/modules/diffusionmodules/model.py:8
↓ 2 callersMethodgt2world_coordinate
x2ct_nerf/modules/INREncoderZoomAxisInAlign.py:68
↓ 2 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
x2ct_nerf/models/base.py:49
↓ 2 callersMethodinitialize
(self, input)
taming/modules/util.py:22
↓ 2 callersFunctionload_config
(config_path, display=False)
main_test_zoom.py:40
↓ 2 callersFunctionload_vqgan
(config, model_module, ckpt_path=None)
main_test_zoom.py:48
↓ 2 callersMethodlog_img
(self, pl_module, batch, batch_idx, split="train")
main.py:365
↓ 2 callersFunctionmain_test
(args)
main_test_zoom.py:399
↓ 2 callersFunctionmd5_hash
(path)
taming/util.py:30
↓ 2 callersFunctionnormalize_tensor
(x,eps=1e-10)
x2ct_nerf/modules/losses/lpips.py:116
↓ 2 callersFunctionnormalize_tensor
(x,eps=1e-10)
taming/modules/losses/lpips.py:116
↓ 2 callersFunctionpsnr
(img_1, img_2, PIXEL_MAX=255.0)
utils/metrics.py:306
↓ 2 callersFunctionsave_scan_mhda
(volume, origin, spacing, path)
data_preprocessing/ctxray_utils.py:38
↓ 2 callersFunctionset_model
(pretrained_weight, weight_dir, model_name)
x2ct_nerf/modules/image_encoder/resnet.py:22
↓ 2 callersMethodset_resolution
(self, full_resolution=False)
x2ct_nerf/modules/INREncoderZoomAxisInAlign.py:58
↓ 2 callersMethodset_resolution
(self, full_resolution=False)
x2ct_nerf/modules/INRGLEncoderZoomAxisInAlign.py:75
↓ 2 callersMethodsetting_base
(self, batch)
x2ct_nerf/models/aegan.py:834
↓ 2 callersMethodsetting_base_val
(self, batch)
x2ct_nerf/models/aegan.py:852
↓ 2 callersMethodto_rgb
(self, x)
x2ct_nerf/models/base.py:138
↓ 2 callersMethodto_rgb
(self, x)
x2ct_nerf/models/aegan.py:196
↓ 2 callersFunctiontruncated_normal_
(tensor, mean=0, std=1)
x2ct_nerf/modules/volume_rendering.py:219
↓ 1 callersMethod__init__
(self)
x2ct_nerf/modules/nerf/model.py:8
↓ 1 callersMethod__init__
(self)
x2ct_nerf/modules/losses/perceptual.py:10
↓ 1 callersMethod__init__
(self, lossconfig, ckpt_path=None, ignore_keys=[],
x2ct_nerf/models/base.py:154
↓ 1 callersMethod__init__
(self)
taming/modules/losses/vqperceptual.py:10
↓ 1 callersMethod_get_nograd_nerf
(self)
x2ct_nerf/modules/INREncoderZoomAxisInAlign.py:196
↓ 1 callersMethod_get_nograd_nerf
(self)
x2ct_nerf/modules/INRGLEncoderZoomAxisInAlign.py:212
↓ 1 callersFunction_isArrayLike
(obj)
data_preprocessing/4_ct_dataset_prepare.py:13
↓ 1 callersMethodapply_preprocessing_xray_according2cam
(self, xray_img, src_camtype)
x2ct_nerf/data/base.py:73
↓ 1 callersFunctionarray2string
(ndarray)
data_preprocessing/2_xraypro.py:33
↓ 1 callersFunctionbatchify
Constructs a version of 'fn' that applies to smaller batches.
x2ct_nerf/modules/nerf/model_utils.py:5
↓ 1 callersMethodcalculate_adaptive_weight
(self, nll_loss, g_loss, last_layer=None)
x2ct_nerf/modules/losses/vqperceptual.py:63
next →1–100 of 506, ranked by callers