Functions — github.com/KingJamesSong/OrientationDiffusion

hub / github.com/KingJamesSong/OrientationDiffusion / functions

Functions151 in github.com/KingJamesSong/OrientationDiffusion

⨍ Functions 151◇ Types & classes 25

↓ 29 callersFunctionsave_images

(images, path, one_channel=False, cmap='jet',**kwargs)

utils.py:30

↓ 16 callersMethod__init__

(self, in_channels, out_channels, emb_dim=256)

modules.py:144

↓ 10 callersFunctionget_data

(args)

utils.py:129

↓ 10 callersFunctionsetup_logging

(run_name)

utils.py:180

↓ 9 callersMethodcoupling_noise

(self, x)

train_orientation_diffusion.py:102

↓ 9 callersMethodcoupling_noise_local

(self, x, M=4)

train_nse.py:288

↓ 8 callersFunctionphase_modulate

(x)

train_orientation_diffusion.py:72

↓ 8 callersFunctionphase_modulate

(x)

train_nse.py:74

↓ 7 callersMethodcoupling_noise_local

(self, x, M=4)

train_nse_cond.py:291

↓ 7 callersMethodnoise_amp

(self, x, t, acc_coherence)

train_nse.py:314

↓ 6 callersMethodclose

(self)

nse_pde.py:65

↓ 6 callersFunctionphase_modulate

(x)

train_nse_cond.py:77

↓ 5 callersFunctionmap_to_image

(mag, phase)

train_nse_cond.py:98

↓ 5 callersFunctionmap_to_image

(x)

train_orientation_diffusion.py:76

↓ 5 callersMethodnoise_amp

(self, x, t, acc_coherence)

train_nse_cond.py:317

↓ 5 callersMethodstep_ema

(self, ema_model, model, step_start_ema=2000)

modules.py:23

↓ 4 callersFunctionangle_space

(score, angle)

train_orientation_diffusion.py:82

↓ 4 callersFunctionmap_to_image

(mag, phase)

train_nse.py:95

↓ 4 callersFunctionsave_mag

(mag)

train_nse.py:84

↓ 4 callersFunctionsave_phase

(phase)

train_nse.py:78

↓ 3 callersFunctionangle_space

(score, angle)

train_nse.py:252

↓ 3 callersMethodcalculate_inception_features

(self, samples)

fid_evaluation.py:52

↓ 3 callersMethodload_or_precalc_dataset_stats

(self)

fid_evaluation.py:64

↓ 3 callersFunctionmap_to_phase

(x)

train_nse_cond.py:69

↓ 3 callersFunctionmap_to_phase

(x)

train_nse.py:68

↓ 3 callersMethodsample

(self, model, n, run_name=None)

train_orientation_diffusion.py:152

↓ 3 callersMethodsample_image

(self, model1, x, amp, model2, x_cond, amp_cond)

train_nse_cond.py:360

↓ 3 callersMethodsample_image

(self, model1, x, amp, model2)

train_nse.py:357

↓ 2 callersFunctionangle_space

(score, angle)

train_nse_cond.py:256

↓ 2 callersFunctionfit_slope_auto

Automatically fit slope of log-log energy spectrum in a dynamic wavenumber range. Args: spectrum: 1D numpy array of E(k) H,

train_nse_cond.py:173

↓ 2 callersFunctionfit_slope_auto

Automatically fit slope of log-log energy spectrum in a dynamic wavenumber range. Args: spectrum: 1D numpy array of E(k) H,

train_nse.py:169

↓ 2 callersFunctionmap_to_phase

(x)

train_orientation_diffusion.py:68

↓ 2 callersMethodnoise_images

(self, x, t)

train_orientation_diffusion.py:129

↓ 2 callersFunctionradial_energy_spectrum

Compute 1D radial energy spectrum from 2D velocity field. Args: vel: Tensor of shape [2, H, W] (vx, vy) Returns: spectrum

train_nse_cond.py:139

↓ 2 callersFunctionradial_energy_spectrum

Compute 1D radial energy spectrum from 2D velocity field. Args: vel: Tensor of shape [2, H, W] (vx, vy) Returns: spectrum

train_nse.py:135

↓ 2 callersFunctiontext_save

(filename, data)

utils.py:14

↓ 1 callersMethodclose

(self)

nse_pde.py:145

↓ 1 callersMethodcompute_score

(self, x_t, x_0, t, acc_coherence)

train_nse_cond.py:351

↓ 1 callersMethodcompute_score

(self, x_t, x_0, t, acc_coherence)

train_nse.py:348

↓ 1 callersFunctionevaluate_dataloaders_mse

(args)

train_nse_cond.py:534

↓ 1 callersFunctionevaluate_dataloaders_spectral

(args, plot=True)

train_nse.py:558

↓ 1 callersMethodfid_score_image

(self)

fid_evaluation.py:119

↓ 1 callersMethodfid_score_noise

(self)

fid_evaluation.py:99

↓ 1 callersFunctionlaunch

()

train_nse_cond.py:572

↓ 1 callersFunctionlaunch

()

train_orientation_diffusion.py:315

↓ 1 callersFunctionlaunch

()

train_nse.py:615

↓ 1 callersFunctionlog_mse_distance

Compute log-MSE between two spectra.

train_nse_cond.py:209

↓ 1 callersFunctionlog_mse_distance

Compute log-MSE between two spectra.

train_nse.py:205

↓ 1 callersMethodnoise_images

(self, x, log_amp_0, t)

train_nse_cond.py:323

↓ 1 callersMethodnoise_images

(self, x, log_amp_0, t)

train_nse.py:320

↓ 1 callersMethodnormalize

(self, data)

dataset.py:68

↓ 1 callersFunctionnum_to_groups

(num, divisor)

fid_evaluation.py:13

↓ 1 callersMethodopen_lmdb

(self)

dataset.py:64

↓ 1 callersFunctionparse_int_list

(s)

dataset.py:11

↓ 1 callersMethodpos_encoding

(self, t, channels)

modules.py:216

↓ 1 callersMethodpos_encoding

(self, t, channels)

modules.py:278

↓ 1 callersMethodpos_encoding

(self, t, channels)

modules.py:340

↓ 1 callersMethodpos_encoding

(self, t, channels)

modules.py:403

↓ 1 callersMethodpos_encoding

(self, t, channels)

modules.py:469

↓ 1 callersMethodpos_encoding

(self, t, channels)

modules.py:534

↓ 1 callersMethodpos_encoding

(self, t, channels)

modules.py:601

↓ 1 callersMethodpos_encoding

(self, t, channels)

modules.py:668

↓ 1 callersMethodpos_encoding

(self, t, channels)

modules.py:735

↓ 1 callersMethodprepare_noise_schedule

(self, noise_start, noise_end)

train_nse_cond.py:279

↓ 1 callersMethodprepare_noise_schedule

(self, noise_start, noise_end)

train_orientation_diffusion.py:99

↓ 1 callersMethodprepare_noise_schedule

(self, noise_start, noise_end)

train_nse.py:276

↓ 1 callersMethodreset_parameters

(self, ema_model, model)

modules.py:31

↓ 1 callersMethodsample_image

(self,model, x)

train_orientation_diffusion.py:170

↓ 1 callersMethodsample_timesteps

(self,n, epoch)

train_nse_cond.py:357

↓ 1 callersMethodsample_timesteps

(self,n, epoch)

train_orientation_diffusion.py:149

↓ 1 callersMethodsample_timesteps

(self,n, epoch)

train_nse.py:354

↓ 1 callersFunctionsave_mag

(mag)

train_nse_cond.py:87

↓ 1 callersFunctionsave_phase

(phase)

train_nse_cond.py:81

↓ 1 callersFunctionspectrum_metrics_batch

Compute slope difference and Wasserstein distance between spectra of real and fake velocity fields. Args: real, fake: torch.Tensor o

train_nse.py:211

↓ 1 callersFunctiontrain_dsm

(args)

train_orientation_diffusion.py:187

↓ 1 callersFunctiontrain_joint

(args)

train_nse_cond.py:410

↓ 1 callersFunctiontrain_joint

(args)

train_nse.py:402

↓ 1 callersMethodupdate_average

(self, old, new)

modules.py:18

↓ 1 callersMethodupdate_model_average

(self, ma_model, current_model)

modules.py:13

↓ 1 callersFunctionvalidate

(args)

train_nse.py:471

↓ 1 callersFunctionvisualize_history_prediction

(args)

train_nse_cond.py:484

↓ 1 callersFunctionwrapped_gaussian_score

Compute the score function for a wrapped Gaussian on [-pi, pi]. Args: y (torch.Tensor): Angles in [-pi, pi], shape [*], can be batch.

train_nse_cond.py:22

↓ 1 callersFunctionwrapped_gaussian_score

Compute the score function for a wrapped Gaussian on [-pi, pi]. Args: y (torch.Tensor): Angles in [-pi, pi], shape [*], can be bat

train_orientation_diffusion.py:21

↓ 1 callersFunctionwrapped_gaussian_score

Compute the score function for a wrapped Gaussian on [-pi, pi]. Args: y (torch.Tensor): Angles in [-pi, pi], shape [*], can be batch.

(self, path, mode="train", normalize=True)

nse_pde.py:11

Method__init__

(self, path, mode="train", normalize=True, dx=1.0, dy=1.0)

nse_pde.py:75

Method__init__

(self, root, resolution=128, raw_resolution=128, num_channe

dataset.py:25

Method__init__

(self, root, transform=None)

dataset.py:77

Method__init__

(self, noise_steps=100, noise_start=1e-4, noise_end=0.1, coupling_start=3e-3, coupling_end=0.045, beta_start=1

train_nse_cond.py:264

(self, channels, size)

modules.py:36

Method__init__

(self, in_channels, out_channels, mid_channels=None, residual=False)

modules.py:59

Method__init__

(self, in_channels, out_channels, mid_channels=None, residual=False)

modules.py:79

Method__init__

(self, in_channels, out_channels, emb_dim=256)

modules.py:100

Method__init__

(self, in_channels, out_channels, emb_dim=256)

modules.py:122

next →1–100 of 151, ranked by callers