MCPcopy Create free account

hub / github.com/alexkeroro86/SampleNeRFRO / functions

Functions490 in github.com/alexkeroro86/SampleNeRFRO

↓ 18 callersMethodupdate
sdf/src/sdf.cpp:384
↓ 13 callersMethodkdtree_get_pt
Returns the dim'th component of the idx'th point in the class:
sdf/include/sdf/internal/nanoflann.hpp:2017
↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
calib/llff/poses/colmap_read_model.py:70
↓ 13 callersMethodsize
sdf/include/sdf/internal/nanoflann.hpp:166
↓ 12 callersMethod_normalize
Normalization helper function.
rnerf/datasets.py:678
↓ 10 callersMethodRead
sdf/include/sdf/internal/RTree.h:424
↓ 10 callersMethodWrite
sdf/include/sdf/internal/RTree.h:412
↓ 10 callersMethoddataset_get
Helper accessor to the dataset points:
sdf/include/sdf/internal/nanoflann.hpp:821
↓ 9 callersMethodIsInternalNode
sdf/include/sdf/internal/RTree.h:298
↓ 9 callersFunctioncolor_space_transform
Transforms inputs between different color spaces :param input_color: tensor of colors to transform (with CxHxW layout) :param fromSpace2toSpace:
metric/flip/flip_api.py:134
↓ 9 callersMethodfindNeighbors
sdf/include/sdf/internal/nanoflann.hpp:1221
↓ 9 callersMethodinit
sdf/include/sdf/internal/nanoflann.hpp:158
↓ 8 callersFunctioncosine_easing_factor
(band, alpha)
rnerf/sh.py:206
↓ 8 callersMethodrun
(self)
rnerf/datasets.py:117
↓ 7 callersMethodaccum_dist
sdf/include/sdf/internal/nanoflann.hpp:363
↓ 6 callersMethodkdtree_get_point_count
Must return the number of data points
sdf/include/sdf/internal/nanoflann.hpp:2012
↓ 6 callersFunctionweighted_percentile
Computes a weighted percentile of an error map, i.e., the error which is such that the sum of the errors in the error map that are smaller than it
metric/flip/flip.py:69
↓ 5 callersMethodcontains
sdf/src/sdf.cpp:379
↓ 5 callersFunctiongaussian_filter
r""" Blur input with 1-D kernel Args: input (torch.Tensor): a batch of tensors to be blured window (torch.Tensor): 1-D gauss kerne
metric/ssim/ssim.py:29
↓ 5 callersFunctionmaybe_parallel_for
sdf/src/util.cpp:7
↓ 5 callersFunctionsave_image
Saves image as png :param img_file: image's filename :param img: float image (in the [0,1] range) to save
metric/flip/data.py:75
↓ 5 callersMethodworstDist
sdf/include/sdf/internal/nanoflann.hpp:204
↓ 4 callersFunctionCHWtoHWC
Transforms an image from CxHxW layout to HxWxC :param x: image with CxHxW layout :return: image with HxWxC layout
metric/flip/data.py:66
↓ 4 callersMethodReadArray
sdf/include/sdf/internal/RTree.h:430
↓ 4 callersMethodWriteArray
sdf/include/sdf/internal/RTree.h:418
↓ 4 callersFunction_fspecial_gauss_1d
r"""Create 1-D gauss kernel Args: size (int): the size of gauss kernel sigma (float): sigma of normal distribution Returns:
metric/ssim/ssim.py:11
↓ 4 callersFunction_normalize
Normalization helper function.
calib/vis_camera_pose_with_llff.py:51
↓ 4 callersFunctionfeature_detection
Detects edges and points (features) in the achromatic image :param imgy: achromatic image (on 1xHxW layout, containing normalized Y-values from YC
metric/flip/flip_api.py:400
↓ 4 callersFunctionhunt_adjustment
Applies Hunt-adjustment to an image :param img: image to adjust (with CxHxW layout in the L*a*b* color space) :return: Hunt-adjusted image (with
metric/flip/flip_api.py:350
↓ 4 callersFunctionminify
(basedir, factors=[], resolutions=[])
calib/llff/poses/pose_utils.py:142
↓ 4 callersFunctionnamedtuple_map
Apply `fn` to each element of `tup` and cast to `tup`'s namedtuple.
rnerf/utils.py:70
↓ 4 callersMethodpeek
Peek at the next training batch or test example without dequeuing it. Returns: batch: dict, has "pixels" and "rays".
rnerf/datasets.py:105
↓ 4 callersFunctionput_text
(img, text, font_color=(0, 0, 0))
metric/export.py:13
↓ 4 callersFunctionsave_err
(fpath, img)
metric/summary.py:43
↓ 4 callersMethodverts_mutable
sdf/src/sdf.cpp:363
↓ 3 callersFunctionHWCtoCHW
Transforms an image from HxWxC layout to CxHxW :param x: image with HxWxC layout :return: image with CxHxW layout
metric/flip/data.py:57
↓ 3 callersMethodIsLeaf
Not a leaf, but a internal node
sdf/include/sdf/internal/RTree.h:301
↓ 3 callersMethodbuildIndex
* Builds the index */
sdf/include/sdf/internal/nanoflann.hpp:1190
↓ 3 callersMethodclear
sdf/include/sdf/internal/nanoflann.hpp:238
↓ 3 callersMethodevalMetric
sdf/include/sdf/internal/nanoflann.hpp:332
↓ 3 callersMethodfreeIndex
Frees the previously-built index. Automatically called within * buildIndex(). */
sdf/include/sdf/internal/nanoflann.hpp:744
↓ 3 callersMethodfull
sdf/include/sdf/internal/nanoflann.hpp:168
↓ 3 callersFunctiongenerate_spatial_filter
Generates spatial contrast sensitivity filters with width depending on the number of pixels per degree of visual angle of the observer :param pix
metric/flip/flip_api.py:275
↓ 3 callersFunctionload_img
[H, W, C], [0, 1]
metric/summary.py:30
↓ 3 callersFunctionload_img
[H, W, C], [0, 1]
metric/compare.py:29
↓ 3 callersMethodrebuild
Rebuild the KD tree from scratch. Call if data updated.
sdf/include/sdf/internal/sdf_util.hpp:47
↓ 3 callersMethodverts
sdf/src/sdf.cpp:362
↓ 2 callersMethodClose
sdf/include/sdf/internal/RTree.h:404
↓ 2 callersMethodInsert
sdf/include/sdf/internal/RTree.h:467
↓ 2 callersMethodOpen
sdf/include/sdf/internal/RTree.h:387
↓ 2 callersMethodPush
Push node and branch onto iteration stack (For internal use only)
sdf/include/sdf/internal/RTree.h:232
↓ 2 callersMethodRemoveAll
sdf/include/sdf/internal/RTree.h:754
↓ 2 callersMethodSearch
sdf/include/sdf/internal/RTree.h:509
↓ 2 callersMethod_compute_grad
(self, grid)
rnerf/datasets.py:315
↓ 2 callersMethod_generate_rays
Generating rays for all images.
rnerf/datasets.py:216
↓ 2 callersMethod_load_renderings
Load images from disk.
rnerf/datasets.py:376
↓ 2 callersMethod_poses_avg
Average poses according to the original NeRF code.
rnerf/datasets.py:660
↓ 2 callersFunction_ssim_per_channel
r""" Calculate ssim index for each color channel of X and Y Args: X (torch.Tensor): images Y (torch.Tensor): images win (t
metric/ssim/ssim.py:45
↓ 2 callersMethod_viewmatrix
Construct lookat view matrix.
rnerf/datasets.py:669
↓ 2 callersMethodaabb
sdf/src/sdf.cpp:349
↓ 2 callersMethodaddPoint
* Called during search to add an element matching the criteria. * @return true if the search should be continued, false if the results are * suf
sdf/include/sdf/internal/nanoflann.hpp:175
↓ 2 callersFunctionassign
sdf/include/sdf/internal/nanoflann.hpp:123
↓ 2 callersFunctioncheck_nans
Checks reference and test images for NaNs and sets NaNs to 0. Depending on verbosity level, warns if NaNs occur :param reference: float tensor :p
metric/flip/flip.py:372
↓ 2 callersMethodcomputeInitialDistances
sdf/include/sdf/internal/nanoflann.hpp:1005
↓ 2 callersFunctioncompute_dssim
[B, C, H, W], [0, 1]
metric/compare.py:97
↓ 2 callersFunctioncompute_flip
[H, W, C], [0, 1]
metric/compare.py:111
↓ 2 callersFunctioncompute_ldrflip
Computes the FLIP error map between two LDR images, assuming the images are observed at a certain number of pixels per degree of visual angle :p
metric/flip/flip_api.py:439
↓ 2 callersFunctioncompute_lpips
[B, C, H, W], [0, 1]
metric/compare.py:104
↓ 2 callersFunctioncompute_psnr
[B, C, H, W], [0, 1]
metric/compare.py:90
↓ 2 callersFunctionconvolve2d
(z, f)
rnerf/vis.py:29
↓ 2 callersMethoddivideTree
* Create a tree node that subdivides the list of vecs from vind[first] * to vind[last]. The routine is called recursively on each sublist. *
sdf/include/sdf/internal/nanoflann.hpp:857
↓ 2 callersMethodfaces
sdf/src/sdf.cpp:353
↓ 2 callersMethodfaces_mutable
sdf/src/sdf.cpp:354
↓ 2 callersMethodfindNeighbors
sdf/include/sdf/internal/nanoflann.hpp:1582
↓ 2 callersFunctionfind_interval
(x)
rnerf/model_utils.py:362
↓ 2 callersFunctionget_fnames
(data_dir, SCENE, log_dir, TEST, NSVF, DATASET, MASK, MIP, STAGE, CROP)
metric/compare.py:46
↓ 2 callersFunctionget_magma_map
Return the magma map, described here: https://bids.github.io/colormap/ :return: matrix of size 256x3 where each row contains a color in the magma
metric/flip/data.py:144
↓ 2 callersFunctionhas_flag
Return a boolean indicating whether a flag name is supported on the specified compiler.
sdf/setup.py:115
↓ 2 callersFunctionhyab
Computes the HyAB distance between reference and test images :param reference: reference image (with CxHxW layout in the standard or Hunt-adjusted
metric/flip/flip_api.py:368
↓ 2 callersFunctionindex2color
Transforms grayscale index map to colors in a color map :param index_map: integer matrix containing integer versions of the FLIP errors, scaled to
metric/flip/data.py:130
↓ 2 callersMethodkdtree_get_bbox
sdf/include/sdf/internal/nanoflann.hpp:2026
↓ 2 callersFunctionlift_gaussian
Lift a Gaussian defined along a curved ray to 3D coordinates.
rnerf/mip.py:35
↓ 2 callersMethodloadIndex_
Loads a previous index from a binary file. * IMPORTANT NOTE: The set of data points is NOT stored in the file, so the * index object must be c
sdf/include/sdf/internal/nanoflann.hpp:1064
↓ 2 callersFunctionload_image_array
Loads an image (in the [0, 255] range) and transforms it into a numpy array and into the [0, 1] range :param img_file: image's filename :return:
metric/flip/data.py:86
↓ 2 callersFunctionopen_file
(pth, mode="r")
rnerf/utils.py:280
↓ 2 callersFunctionproject_2d
Args - pts: np.ndarray, [H, W, D, 4] - cam_mat: np.ndarray, [4, 4] - view_mat: np.ndarray, [4, 4] Returns: - uv: np.ndarray, [H,
calib/make_visual_hull.py:30
↓ 2 callersFunctionread_exr
Read color data from EXR image file. Set negative values and nans to 0. :param filename: string describing file path :return: RGB image in float3
metric/flip/data.py:99
↓ 2 callersFunctionresize
sdf/include/sdf/internal/nanoflann.hpp:103
↓ 2 callersFunctionrodrigues
sdf/demo.cpp:27
↓ 2 callersFunctionsafe_trig_helper
(x, fn, t=100 * jnp.pi)
rnerf/math_utils.py:28
↓ 2 callersMethodsaveIndex_
Stores the index in a binary file. * IMPORTANT NOTE: The set of data points is NOT stored in the file, so when * loading the index object it m
sdf/include/sdf/internal/nanoflann.hpp:1050
↓ 2 callersFunctionsave_img
(fpath, img)
metric/compare.py:39
↓ 2 callersFunctionspatial_filter
Filters an image with channel specific spatial contrast sensitivity functions and clips result to the unit cube in linear RGB :param img: image t
metric/flip/flip_api.py:326
↓ 2 callersFunctionto_trans_quat
calib/vis_camera_pose_with_llff.py:77
↓ 2 callersFunctionto_view_matrix
Args: - mat: np.ndarray, [4, 4] Returns: - ret: np.ndarray, [4, 4]
calib/make_visual_hull.py:18
↓ 2 callersFunctiontone_map
Applies exposure compensation and tone mapping. Refer to the Visualizing Errors in Rendered High Dynamic Range Images paper for details about the
metric/flip/flip_api.py:65
↓ 2 callersFunctiontree_sum_fn
(fn)
train.py:147
↓ 2 callersFunctionvisualize_depth
Visualize a depth map. Args: depth: A depth map. acc: An accumulation map, in [0, 1]. near: The depth of the near plane, if None then j
rnerf/vis.py:45
↓ 1 callersMethodInit
Reset iterator
sdf/include/sdf/internal/RTree.h:190
↓ 1 callersMethodOpenRead
sdf/include/sdf/internal/RTree.h:396
next →1–100 of 490, ranked by callers