MCPcopy Create free account

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

Functions194 in github.com/computational-imaging/bacon

↓ 13 callersMethod__init__
(self, w0=20)
modules.py:69
↓ 9 callersMethodtoggle_logging_sampling
(self)
dataio.py:484
↓ 8 callersFunctiondict2cuda
(a_dict)
training.py:168
↓ 8 callersFunctionprocess_batch_in_chunks
(in_dict, model, max_chunk_size=1024, progress=None)
utils.py:695
↓ 7 callersFunctionpeak_signal_noise_ratio
Calculate PSNR using GT and predicted image (assumes valid values between 0 and 1
utils.py:768
↓ 6 callersFunctionwrite_psnr
(pred_img, gt_img, writer, iter, prefix)
utils.py:198
↓ 5 callersFunctionlayer_factory
(layer_type, w0=30)
modules.py:649
↓ 4 callersMethodget_camera_params
(self)
dataio.py:423
↓ 3 callersFunctionfont_size
Size of all texts shown in plots
experiments/figure_setup.py:32
↓ 3 callersFunctionmake_contour_plot
(array_2d, mode='log', ax=None)
utils.py:556
↓ 3 callersFunctionto_numpy
(x)
utils.py:18
↓ 2 callersMethod__init__
( self, hidden_size, out_size, n_layers, weight_scale, bias=True, output_act=False )
experiments/plot_activation_distributions.py:30
↓ 2 callersFunctioncompute_one_scale
(model, layer_ind, render_coords, sdf_values, hash_ind)
experiments/render_sdf.py:110
↓ 2 callersFunctioneval_nerf_bacon
(scene, config, checkpoint, outdir, res, scale, chunk_size=10000, return_all=False, val_idx=None)
experiments/render_nerf.py:291
↓ 2 callersMethodget_img_shape
(self)
dataio.py:420
↓ 2 callersFunctionoverfit_sdf_loss_total
fit sdf to sphere via mse loss
loss_functions.py:122
↓ 2 callersFunctionplot_initialization
(use_original=False, plot_fit=True)
experiments/plot_activation_distributions.py:100
↓ 2 callersFunctionplot_samples
(out_dict, num_rays_to_visu=10, xlim=(0, 6))
utils.py:651
↓ 2 callersFunctionrender_in_chunks
(in_dict, model, chunk_size, return_all=False)
experiments/render_nerf.py:124
↓ 2 callersMethodshuffle_rays
(self)
dataio.py:510
↓ 2 callersFunctionsubsample_dict
(in_dict, num_views)
utils.py:774
↓ 2 callersFunctionticks_size
Size of axes' ticks
experiments/figure_setup.py:38
↓ 2 callersFunctionwrite_sdf_slice
write slices of sdf in each plane
utils.py:583
↓ 1 callersMethod__init__
(self, pointcloud_path, num_samples=30**3, coarse_scale=1e-1, fine_scale=1e-3)
dataio.py:634
↓ 1 callersMethod_generate_rays_normalized
(self)
dataio.py:492
↓ 1 callersMethod_precompute_rays
(self)
dataio.py:513
↓ 1 callersFunctionaxis_lw
Line width of the axes
experiments/figure_setup.py:44
↓ 1 callersFunctioncompute_one_scale_adaptive
(model, layer_ind, render_coords, sdf_values, hash_ind, threshold=0.003)
experiments/render_sdf.py:119
↓ 1 callersFunctioncond_mkdir
(path)
utils.py:22
↓ 1 callersFunctionconvert_to_nerfdata
Convert Blender data to multiscale.
convert_blender_data.py:50
↓ 1 callersFunctioncoord2ind
(xyz_coord, res)
experiments/render_sdf.py:85
↓ 1 callersFunctioncrop_center
(pil_img, crop_width, crop_height)
dataio.py:177
↓ 1 callersFunctioncrop_max_square
(pil_img)
dataio.py:185
↓ 1 callersFunctioncumprod_exclusive
(tensor, dim=-2)
forward_models.py:4
↓ 1 callersFunctionexpected_sin
Estimates mean and variance of sin(z), z ~ N(x, var).
modules.py:518
↓ 1 callersFunctionexport_meshes
(adaptive=True)
experiments/render_sdf.py:176
↓ 1 callersFunctionexport_model
(ckpt_path, model_name, N=512, model_type='bacon', hidden_layers=8, hidden_size=256, output_l
experiments/render_sdf.py:16
↓ 1 callersFunctionexport_model
(ckpt_path, model_name, N=512, model_type='bacon', hidden_layers=8, hidden_size=256, output_l
spectrum_visualization/get_shape_spectra.py:26
↓ 1 callersFunctionextract_spectrum
()
spectrum_visualization/get_shape_spectra.py:116
↓ 1 callersMethodforward_mfn
(self, input_dict)
modules.py:202
↓ 1 callersFunctiongenerate_mesh
(model, N, return_sdf=False, num_outputs=4, model_name='model')
experiments/render_sdf.py:48
↓ 1 callersFunctiongenerate_mesh_adaptive
(model, model_name)
experiments/render_sdf.py:139
↓ 1 callersFunctionget_mgrid
Generates a flattened grid of (x,y,...) coordinates in a range of -1 to 1.
dataio.py:18
↓ 1 callersMethodget_rays
(self, idx)
dataio.py:563
↓ 1 callersFunctionget_ref_spectrum
(xyz_file, N)
spectrum_visualization/get_shape_spectra.py:90
↓ 1 callersMethodget_samples
(self, range, sampling_density)
dataio.py:71
↓ 1 callersMethodget_val_rays
(self)
dataio.py:546
↓ 1 callersFunctioninit_dataloader
load sdf dataloader via eikonal equation or fitting sdf directly
experiments/train_sdf.py:126
↓ 1 callersFunctioninit_dataloader
load image datasets, dataloader
experiments/train_img.py:119
↓ 1 callersFunctioninit_loss
define loss, summary functions given expmt configs
experiments/train_sdf.py:214
↓ 1 callersFunctioninit_loss
define loss, summary functions given expmt configs
experiments/train_img.py:202
↓ 1 callersFunctioninit_model
return appropriate model given experiment configs
experiments/train_sdf.py:141
↓ 1 callersFunctioninit_model
(opt)
experiments/train_img.py:148
↓ 1 callersFunctioninit_multiscale_mc
()
experiments/render_sdf.py:192
↓ 1 callersFunctionintegrated_pos_enc
Encode `x` with sinusoids scaled by 2^[min_deg:max_deg-1].
modules.py:526
↓ 1 callersFunctionlabel_size
Size of axis labels
experiments/figure_setup.py:26
↓ 1 callersMethodlayer_forward
for multiscale SDF extraction
modules.py:302
↓ 1 callersFunctionload_dataset
(opt, res, scale)
experiments/render_nerf.py:72
↓ 1 callersMethodload_images
(self, s, meta, testskip)
dataio.py:351
↓ 1 callersMethodload_mesh
(self, pointcloud_path)
dataio.py:647
↓ 1 callersFunctionload_model
(opt, checkpoint)
experiments/render_nerf.py:95
↓ 1 callersFunctionload_renderings
Load images and metadata from disk.
convert_blender_data.py:21
↓ 1 callersFunctionlr_log_schedule
(it, num_steps=1e6, nw=2500, lr0=1e-3, lrn=5e-6, lambdaw=0.01)
training.py:202
↓ 1 callersFunctionmain
()
experiments/train_1d.py:63
↓ 1 callersFunctionmain
()
experiments/train_sdf.py:91
↓ 1 callersFunctionmain
()
experiments/train_img.py:84
↓ 1 callersFunctionmain
()
experiments/train_radiance_field.py:77
↓ 1 callersFunctionmake_square_axes
Make an axes square in screen units. Should be called after plotting.
experiments/plot_activation_distributions.py:15
↓ 1 callersMethodnet_forward
(self, coords)
modules.py:429
↓ 1 callersFunctionnormalize
(coords, scaling=0.9)
spectrum_visualization/get_shape_spectra.py:75
↓ 1 callersMethodnormalize
(self, coords)
dataio.py:659
↓ 1 callersFunctionplot_lw
Line width of the plotted curves
experiments/figure_setup.py:50
↓ 1 callersMethodpose_spherical
(self, theta, phi, radius)
dataio.py:410
↓ 1 callersFunctionprepare_multi_scale
(res, num_scales)
experiments/render_sdf.py:84
↓ 1 callersFunctionreg_schedule
(it, num_steps=1e6, lr0=1e-3, lrn=1e-4)
training.py:198
↓ 1 callersFunctionrender_all_in_chunks
(in_dict, model, scale, chunk_size)
experiments/render_nerf.py:157
↓ 1 callersFunctionrender_image
(opt, models, dataset, chunk_size, in_dict, meta_dict, gt_dict, scale, retur
experiments/render_nerf.py:195
↓ 1 callersMethodrotate_by_phi_along_x
(self, phi)
dataio.py:396
↓ 1 callersMethodrotate_by_theta_along_y
(self, theta)
dataio.py:403
↓ 1 callersFunctionsample_pdf
hierarchical sampling code for neural radiance fields
training.py:410
↓ 1 callersMethodsample_surface
(self)
dataio.py:667
↓ 1 callersFunctionsave_params
(opt, model)
experiments/train_sdf.py:229
↓ 1 callersFunctionsave_params
(opt, model)
experiments/train_img.py:218
↓ 1 callersFunctionsubdiv_index
(xyz_prev, next_res)
experiments/render_sdf.py:93
↓ 1 callersFunctiontrain
()
experiments/train_1d.py:72
↓ 1 callersFunctiontrain
()
experiments/train_sdf.py:100
↓ 1 callersFunctiontrain
()
experiments/train_img.py:93
↓ 1 callersFunctiontrain
(validation=True)
experiments/train_radiance_field.py:84
↓ 1 callersMethodtranslate_by_t_along_z
(self, t)
dataio.py:391
FunctionIntegratedPositionalEncoding
(coords, radius, L=8)
modules.py:515
Method__enter__
(self)
utils.py:779
Method__enter__
(self)
spectrum_visualization/get_shape_spectra.py:17
Method__exit__
(self, exc_type, exc_val, exc_tb)
utils.py:783
Method__exit__
(self, exc_type, exc_val, exc_tb)
spectrum_visualization/get_shape_spectra.py:21
Method__getitem__
(self, idx)
dataio.py:84
Method__getitem__
(self, idx)
dataio.py:157
Method__getitem__
(self, idx)
dataio.py:226
Method__getitem__
(self, item)
dataio.py:432
Method__getitem__
(self, idx)
dataio.py:581
Method__getitem__
(self, idx)
dataio.py:685
next →1–100 of 194, ranked by callers