Functions — github.com/computational-imaging/bacon

hub / github.com/computational-imaging/bacon / functions

Functions194 in github.com/computational-imaging/bacon

⨍ Functions 194◇ Types & classes 26

Method__init__

(self, range, fn, grad_fn=None, sampling_density=100, train_every=10)

dataio.py:60

Method__init__

(self, dataset, compute_diff='all', centered=True, include_end=False, multiscale=False, stage

dataio.py:120

Method__init__

(self, filename, grayscale=False, resolution=None, root_path=None, crop_square=True, url=None

dataio.py:190

Method__init__

(self, basedir, mode='train', splits=['train', 'val', 'test'], select_idx=No

dataio.py:271

Method__init__

(self, dataset, img_shape, camera_params, samples_per_ray=128, samples_per_v

(self, hidden_size, out_size, n_layers, weight_scale, bias=True, output_act=False)

modules.py:104

Method__init__

(self, in_features, out_features, weight_scale, quantization_interval=2*np.pi)

modules.py:138

Method__init__

(self, in_size, hidden_size, out_size, hid

modules.py:162

Method__init__

(self, in_size, hidden_size, out_size, hid

modules.py:244

Method__init__

(self, out_features=1, nl='sine', in_features=1, hidden_features=256, num_hidden_layers=3,

modules.py:393

Method__init__

(self, out_features=1, nl='sine', in_features=1, hidden_features=256, num_hidden_layers=3,

modules.py:469

Method__init__

(self, embedding_size, scale, dims=2, gaussian=True)

modules.py:546

Method__init__

(self, num_encoding_functions=6, include_input=True, log_sampling=True, normalize=False, inpu

modules.py:584

Method__init__

(self, in_features, out_features, num_hidden_layers, hidden_features, outerm

modules.py:671

Method__init__

(self, in_features=2, out_features=1, hidden_features=256, num_hidden_layers=3, w0=30,

modules.py:776

Method__init__

(self, **kwargs)

experiments/render_nerf.py:60

Method__init__

(self, in_features, out_features, weight_scale)

experiments/plot_activation_distributions.py:66

Method__init__

( self, in_size, hidden_size, out_size, n_layers=3, input_scal

experiments/plot_activation_distributions.py:78

Functionchunk_lists_from_batch_reduce_to_raysamples_fn

(model_input, meta, gt, max_chunk_size)

Functioncompute_disp_from_depth

(pred_depth, pred_weights)

forward_models.py:35

Functioncompute_tomo_depth

(pred_weights, zs)

forward_models.py:30

Functioncompute_tomo_radiance

(pred_weights, pred_rgb, black_background=False)

forward_models.py:19

Functioncompute_transmittance_weights

(pred_sigma, t_intervals)

convert_blender_data.py:45

Functionfigure_setup

Set all the sizes to the correct values and use tex fonts for all texts.

experiments/figure_setup.py:56

Functionfirst_layer_sine_init

(self, model_input, mode=None, integral_dim=None)

modules.py:232

Methodforward

(self, model_input, specified_layers=None, get_feature=False, continue_layer=0, continue_featu

Apply positional encoding to the input.

modules.py:574

Methodforward

r"""Apply positional encoding to the input. Args: tensor (torch.Tensor): Input tensor to be positionally encoded. enc

experiments/plot_activation_distributions.py:49

Methodforward

(self, x)

experiments/plot_activation_distributions.py:73

(coords, sigma=1, center=0.5)

dataio.py:94

Functionget_fig_size

Convert dimensions in centimeters to inches. If no height is given, it is computed using the golden ratio.

experiments/figure_setup.py:8

Methodget_num_samples

Functioninit_weights_elu

(m)

modules.py:26

Functioninit_weights_normal

(m)

modules.py:13

Functioninit_weights_selu

(m)

modules.py:19

Functioninit_weights_uniform

(m)

modules.py:39

Functioninit_weights_xavier

(m)

modules.py:33

Functionlin2img

(tensor, image_resolution=None)

convert_blender_data.py:74

Functionmfn_weights_init

(m)

modules.py:95

Functionmultiscale_image_mse

(model_output, gt, use_resized=False)

loss_functions.py:23

Functionmultiscale_overfit_sdf

(model_output, gt, coarse_loss_weight=1e-2)

loss_functions.py:117

Functionmultiscale_radiance_loss

(model_outputs, gt, use_resized=False, weight=1.0, regularize_sigma=False, reg_la

loss_functions.py:34

Methodmultiscale_resize

(model_output, gt, coarse_loss_weight=1e-2)

loss_functions.py:112

Functionpolynomial_1

(coords)

dataio.py:102

Functionradiance_sigma_rgb_loss

(model_outputs, gt, regularize_sigma=False, reg_lambda=1e-5, reg_c=0.5)

Functionsampling_scheduler

(step, start=0, lr0=1e-4, lrn=1e-4)

training.py:31

Functionsave_fig

Save a Matplotlib figure as EPS/PNG/PDF to the given path and trim it.

experiments/figure_setup.py:94

Functionsave_img

given np array, convert to image and save

(model, train_dataloader, steps, lr, steps_til_summary, steps_til_checkpoint, model_dir, loss_fn, su

training.py:13

Functiontrain_wchunks

(models, train_dataloader, num_steps, lr, steps_til_summary, steps_til_checkpoint, model_dir,

training.py:207

Functionwrite_image_summary

(image_resolution, train_dataset, model, model_input, gt, model_output, writer, total_

utils.py:158

Functionwrite_multiscale_image_summary

(image_resolution, train_dataset, model, model_input, gt, model_output, wri

utils.py:69

Functionwrite_multiscale_radiance_summary

(models, train_dataloader, val_dataloader, loss_fn, optims, meta, gt, wr

utils.py:224

Functionwrite_multiscale_sdf_summary

(model, model_input, gt, model_output, writer, total_steps, prefix='train_')

utils.py:641

Functionwrite_radiance_summary

(models, train_dataloader, val_dataloader, loss_fn, optims, meta, gt, writer, total

utils.py:429

Functionwrite_sdf_summary

(model, model_input, gt, model_output, writer, total_steps, prefix='train_')

utils.py:646

Functionwrite_simple_1D_function_summary

(dataset, model, model_input, gt, model_output, writer, total_steps, prefix='train')

← previous101–194 of 194, ranked by callers