Index

_

__init__() (nndt.primitive_sdf.SphereSDF method) (nndt.space2.abstracts.AbstractBBoxNode method) (nndt.space2.abstracts.AbstractTreeElement method) (nndt.vizualize.BasicVizualization method)

_vec_scale_rotate_shift() (in module nndt.datagen)

A

AbstractBBoxNode (class in nndt.space2.abstracts) AbstractDatagen (class in nndt.datagen) AbstractLoader (class in nndt.space2.abstracts) AbstractSDF (class in nndt.primitive_sdf) AbstractTrainableTask (class in nndt.trainable_task) AbstractTransformation (class in nndt.space2.transformation) AbstractTreeElement (class in nndt.space2.abstracts) ACDC (class in nndt.datasets.acdc)

add_sphere() (in module nndt.space2.simple_api) ApproximateSDF (class in nndt.trainable_task) ApproximateSDF.DATA (class in nndt.trainable_task) ApproximateSDFLipMLP (class in nndt.trainable_task) ApproximateSDFLipMLP.DATA (class in nndt.trainable_task) ApproximateSDFLipMLP2 (class in nndt.trainable_task) ApproximateSDFLipMLP2.DATA (class in nndt.trainable_task) array_to_vert_and_faces() (in module nndt.space2.utils)

B

barycentric_grid() (in module nndt.math_core) BasicVizualization (class in nndt.vizualize) bbox (nndt.primitive_sdf.AbstractSDF property) (nndt.primitive_sdf.SphereSDF property) (nndt.space2.implicit_representation.IR1SDF property) (nndt.space2.loader.IR1Loader property)

C

calc_bbox() (nndt.space2.abstracts.AbstractLoader method) (nndt.space2.loader.IR1Loader method) (nndt.space2.loader.MeshObjLoader method) (nndt.space2.loader.SDTLoader method)

calc_ret_shape() (in module nndt.space2.utils) CLASS (nndt.trainable_task.SurfaceSegmentation.DATA attribute) ColorMethodSetNode (class in nndt.space2.method_set)

D

DataGenForSegmentation (class in nndt.datagen) DataGenForShapeRegression (class in nndt.datagen) Dataset (class in nndt.datasets.dataset)

dataset_list() (nndt.datasets.dataset.Dataset method) DefaultPreloader (class in nndt.space2.space_preloader) DescConv (class in nndt.haiku_modules) draw_loss() (nndt.vizualize.BasicVizualization method)

E

Eikonal3D (class in nndt.trainable_task) Eikonal3D.DATA (class in nndt.trainable_task)

Eikonal3D.FUNC (class in nndt.trainable_task) EmptyLoader (class in nndt.space2.loader)

F

FileSource (class in nndt.space2.filesource) fix_file_extension() (in module nndt.space2.utils) from_json() (in module nndt.space2.simple_api) fun (nndt.primitive_sdf.AbstractSDF property)

fun2vec_and_grad() (in module nndt.primitive_sdf) FUNC (nndt.trainable_task.ApproximateSDF attribute) (nndt.trainable_task.ApproximateSDFLipMLP attribute) (nndt.trainable_task.ApproximateSDFLipMLP2 attribute) functions (nndt.space2.loader.IR1Loader property)

G

get() (nndt.datagen.AbstractDatagen method) (nndt.datagen.DataGenForSegmentation method) (nndt.datagen.DataGenForShapeRegression method) get_lipschitz_loss() (nndt.haiku_modules.LipLinear method) (nndt.haiku_modules.LipMLP method)

grid_in_cube() (in module nndt.math_core) grid_in_cube2() (in module nndt.math_core) Group (class in nndt.space2.group)

H

help_barycentric_grid() (in module nndt.math_core)

I

IdentityTransform (class in nndt.space2.transformation) ImpRepr (class in nndt.space2.implicit_representation) init() (nndt.space2.space.Space method) init_and_functions() (nndt.trainable_task.AbstractTrainableTask method) (nndt.trainable_task.ApproximateSDF method) (nndt.trainable_task.ApproximateSDFLipMLP method) (nndt.trainable_task.ApproximateSDFLipMLP2 method) (nndt.trainable_task.Eikonal3D method) (nndt.trainable_task.SimpleSDF method) (nndt.trainable_task.SurfaceSegmentation method) init_code() (in module nndt.global_config) init_colab() (in module nndt.global_config) init_data() (nndt.trainable_task.AbstractTrainableTask method) (nndt.trainable_task.ApproximateSDF method) (nndt.trainable_task.ApproximateSDFLipMLP method) (nndt.trainable_task.ApproximateSDFLipMLP2 method) (nndt.trainable_task.Eikonal3D method) (nndt.trainable_task.SurfaceSegmentation method)

init_jupyter() (in module nndt.global_config) IR1Loader (class in nndt.space2.loader) IR1SDF (class in nndt.space2.implicit_representation) is_load() (nndt.space2.abstracts.AbstractLoader method) (nndt.space2.loader.EmptyLoader method) (nndt.space2.loader.IR1Loader method) (nndt.space2.loader.MeshObjLoader method) (nndt.space2.loader.SDTLoader method) (nndt.space2.loader.TXTLoader method) is_print_on_epoch() (nndt.vizualize.BasicVizualization method) iter() (nndt.vizualize.BasicVizualization method) IterAccessMixin (class in nndt.space2.abstracts) IteratorWithTimeMeasurements (class in nndt.vizualize)

J

json (nndt.space2.loader.IR1Loader property)

K

kdtree (nndt.space2.loader.MeshObjLoader property)

L

LipLinear (class in nndt.haiku_modules) LipMLP (class in nndt.haiku_modules) load() (nndt.datasets.dataset.Dataset method) load_batch() (nndt.space2.method_set_train_task.TrainTaskSetNode method) load_data() (nndt.space2.abstracts.AbstractLoader method) (nndt.space2.loader.EmptyLoader method) (nndt.space2.loader.IR1Loader method) (nndt.space2.loader.MeshObjLoader method) (nndt.space2.loader.SDTLoader method) (nndt.space2.loader.TXTLoader method)

load_from_file_lists() (in module nndt.space2.simple_api) load_from_path() (in module nndt.space2.simple_api) load_implicit_ir1() (in module nndt.space2.simple_api) load_mesh_obj() (in module nndt.space2.simple_api) load_only_one_file() (in module nndt.space2.simple_api) load_sdt() (in module nndt.space2.simple_api) load_txt() (in module nndt.space2.simple_api)

M

main_loss (nndt.trainable_task.Eikonal3D.FUNC attribute) (nndt.trainable_task.SurfaceSegmentation.FUNC attribute) mesh (nndt.space2.loader.MeshObjLoader property) MeshObjLoader (class in nndt.space2.loader) MeshObjMethodSetNode (class in nndt.space2.method_set) MethodNode (class in nndt.space2.method_set) MethodSetNode (class in nndt.space2.method_set) metric_accuracy (nndt.trainable_task.SurfaceSegmentation.FUNC attribute) MICCAI2015 (class in nndt.datasets.miccai2015) module nndt nndt.datagen nndt.datasets nndt.datasets.acdc nndt.datasets.dataset nndt.datasets.miccai2015 nndt.datasets.utils nndt.global_config nndt.haiku_modules nndt.math_core nndt.primitive_sdf nndt.space2 nndt.space2.abstracts nndt.space2.filesource nndt.space2.group nndt.space2.implicit_representation nndt.space2.loader nndt.space2.method_set nndt.space2.method_set_train_task nndt.space2.object3D nndt.space2.plot_tree nndt.space2.print_tree nndt.space2.simple_api nndt.space2.space nndt.space2.space_preloader nndt.space2.transformation nndt.space2.tree_utils nndt.space2.utils nndt.trainable_task nndt.vizualize

N

nn (nndt.trainable_task.Eikonal3D.FUNC attribute) (nndt.trainable_task.SurfaceSegmentation.FUNC attribute) nn_dx (nndt.trainable_task.Eikonal3D.FUNC attribute) nn_dy (nndt.trainable_task.Eikonal3D.FUNC attribute) nn_dz (nndt.trainable_task.Eikonal3D.FUNC attribute) nndt module nndt.datagen module nndt.datasets module nndt.datasets.acdc module nndt.datasets.dataset module nndt.datasets.miccai2015 module nndt.datasets.utils module nndt.global_config module nndt.haiku_modules module nndt.math_core module nndt.primitive_sdf module nndt.space2 module nndt.space2.abstracts module nndt.space2.filesource module

nndt.space2.group module nndt.space2.implicit_representation module nndt.space2.loader module nndt.space2.method_set module nndt.space2.method_set_train_task module nndt.space2.object3D module nndt.space2.plot_tree module nndt.space2.print_tree module nndt.space2.simple_api module nndt.space2.space module nndt.space2.space_preloader module nndt.space2.transformation module nndt.space2.tree_utils module nndt.space2.utils module nndt.trainable_task module nndt.vizualize module node_method() (in module nndt.space2.abstracts)

O

Object3D (class in nndt.space2.object3D)

P

P (nndt.trainable_task.ApproximateSDF.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP2.DATA attribute) pad_bbox() (in module nndt.space2.utils) params (nndt.space2.loader.IR1Loader property) plot() (nndt.space2.abstracts.AbstractBBoxNode method) points (nndt.space2.loader.MeshObjLoader property) preload() (nndt.space2.space.Space method) (nndt.space2.space_preloader.DefaultPreloader method)

print() (nndt.space2.abstracts.AbstractBBoxNode method) PrintContainer (class in nndt.space2.print_tree) purefun_sdf() (nndt.space2.implicit_representation.ImpRepr method) purefun_vec_sdf() (nndt.space2.implicit_representation.ImpRepr method) purefun_vec_sdf_dx() (nndt.space2.implicit_representation.ImpRepr method) purefun_vec_sdf_dy() (nndt.space2.implicit_representation.ImpRepr method) purefun_vec_sdf_dz() (nndt.space2.implicit_representation.ImpRepr method)

R

read_space_from_file() (in module nndt.space2.simple_api) record() (nndt.vizualize.BasicVizualization method) request() (nndt.primitive_sdf.AbstractSDF method) (nndt.space2.loader.SDTLoader method)

rgba (nndt.space2.loader.MeshObjLoader property) rotation_matrix() (in module nndt.math_core)

S

sampling_eachN_from_mesh() (nndt.space2.method_set.MeshObjMethodSetNode method) sampling_grid() (nndt.space2.method_set.SamplingMethodSetNode method) sampling_grid_with_noise() (nndt.space2.method_set.SamplingMethodSetNode method) sampling_uniform() (nndt.space2.method_set.SamplingMethodSetNode method) SamplingMethodSetNode (class in nndt.space2.method_set) save_3D_array() (nndt.vizualize.BasicVizualization method) save_3D_slices() (in module nndt.vizualize) save_mesh() (nndt.space2.method_set.MeshObjMethodSetNode method) (nndt.vizualize.BasicVizualization method) save_sdt_as_obj() (in module nndt.vizualize) save_space_to_file() (in module nndt.space2.simple_api) (nndt.space2.space.Space method) save_state() (nndt.vizualize.BasicVizualization method) save_txt() (nndt.vizualize.BasicVizualization method) save_verts_and_faces_to_obj() (in module nndt.space2.utils) scale_xyz() (in module nndt.math_core) SDF (nndt.trainable_task.ApproximateSDF.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP2.DATA attribute) (nndt.trainable_task.SimpleSDF.DATA attribute) sdf (nndt.trainable_task.SimpleSDF.FUNC attribute) SDF_CUBE (nndt.trainable_task.SurfaceSegmentation.DATA attribute) sdf_dx (nndt.trainable_task.SimpleSDF.FUNC attribute) sdf_dy (nndt.trainable_task.SimpleSDF.FUNC attribute) sdf_dz (nndt.trainable_task.SimpleSDF.FUNC attribute)

sdt (nndt.space2.loader.SDTLoader property) sdt_to_obj() (nndt.vizualize.BasicVizualization method) SDTLoader (class in nndt.space2.loader) SDTMethodSetNode (class in nndt.space2.method_set) shift_xyz() (in module nndt.math_core) ShiftAndScaleTransform (class in nndt.space2.transformation) SimpleSDF (class in nndt.trainable_task) SimpleSDF.DATA (class in nndt.trainable_task) SimpleSDF.FUNC (class in nndt.trainable_task) Space (class in nndt.space2.space) SphereSDF (class in nndt.primitive_sdf) split_node_kfold() (in module nndt.space2.simple_api) split_node_namelist() (in module nndt.space2.simple_api) split_node_test_train() (in module nndt.space2.simple_api) surface_ind2rgba() (nndt.space2.method_set.ColorMethodSetNode method) surface_ind2xyz() (nndt.space2.method_set.MeshObjMethodSetNode method) surface_rgba() (nndt.space2.method_set.ColorMethodSetNode method) surface_xyz() (nndt.space2.method_set.MeshObjMethodSetNode method) surface_xyz2ind() (nndt.space2.method_set.MeshObjMethodSetNode method) surface_xyz2localsdt() (nndt.space2.method_set.SDTMethodSetNode method) surface_xyz2rgba() (nndt.space2.method_set.ColorMethodSetNode method) surface_xyz2sdt() (nndt.space2.method_set.SDTMethodSetNode method) SurfaceSegmentation (class in nndt.trainable_task) SurfaceSegmentation.DATA (class in nndt.trainable_task) SurfaceSegmentation.FUNC (class in nndt.trainable_task)

T

T (nndt.trainable_task.ApproximateSDF.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP2.DATA attribute) take_each_n() (in module nndt.math_core) text (nndt.space2.loader.TXTLoader property) to_json() (in module nndt.space2.simple_api) (nndt.space2.space.Space method) ToNormalCubeTransform (class in nndt.space2.transformation) train_task_sdt2sdf() (nndt.space2.method_set_train_task.TrainTaskSetNode method) train_test_split() (in module nndt.math_core) TrainTaskSetNode (class in nndt.space2.method_set_train_task) transform_sdt_ns2ps() (nndt.space2.transformation.AbstractTransformation method) (nndt.space2.transformation.IdentityTransform method) (nndt.space2.transformation.ShiftAndScaleTransform method) (nndt.space2.transformation.ToNormalCubeTransform method)

transform_sdt_ps2ns() (nndt.space2.transformation.AbstractTransformation method) (nndt.space2.transformation.IdentityTransform method) (nndt.space2.transformation.ShiftAndScaleTransform method) (nndt.space2.transformation.ToNormalCubeTransform method) transform_xyz_ns2ps() (nndt.space2.transformation.AbstractTransformation method) (nndt.space2.transformation.IdentityTransform method) (nndt.space2.transformation.ShiftAndScaleTransform method) (nndt.space2.transformation.ToNormalCubeTransform method) transform_xyz_ps2ns() (nndt.space2.transformation.AbstractTransformation method) (nndt.space2.transformation.IdentityTransform method) (nndt.space2.transformation.ShiftAndScaleTransform method) (nndt.space2.transformation.ToNormalCubeTransform method) TXTLoader (class in nndt.space2.loader)

U

uniform_in_cube() (in module nndt.math_core) unload_data() (nndt.space2.abstracts.AbstractLoader method) (nndt.space2.loader.EmptyLoader method) (nndt.space2.loader.IR1Loader method) (nndt.space2.loader.MeshObjLoader method) (nndt.space2.loader.SDTLoader method) (nndt.space2.loader.TXTLoader method)

unload_from_memory() (nndt.space2.abstracts.AbstractBBoxNode method) update_bbox() (in module nndt.space2.utils) update_bbox_from_children() (in module nndt.space2.tree_utils) update_bbox_with_float_over_tree() (in module nndt.space2.tree_utils)

V

vec_fun (nndt.primitive_sdf.AbstractSDF property) vec_fun_dx (nndt.primitive_sdf.AbstractSDF property) vec_fun_dy (nndt.primitive_sdf.AbstractSDF property) vec_fun_dz (nndt.primitive_sdf.AbstractSDF property)

vec_main_loss (nndt.trainable_task.SimpleSDF.FUNC attribute) vec_sdf (nndt.trainable_task.SimpleSDF.FUNC attribute) vec_sdf_dx (nndt.trainable_task.SimpleSDF.FUNC attribute) vec_sdf_dy (nndt.trainable_task.SimpleSDF.FUNC attribute) vec_sdf_dz (nndt.trainable_task.SimpleSDF.FUNC attribute)

W

WEIGHT (nndt.trainable_task.ApproximateSDFLipMLP2.DATA attribute)

weight_normalization() (nndt.haiku_modules.LipLinear method)

X

X (nndt.trainable_task.ApproximateSDF.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP2.DATA attribute) (nndt.trainable_task.Eikonal3D.DATA attribute) (nndt.trainable_task.SimpleSDF.DATA attribute)

Y

Y (nndt.trainable_task.ApproximateSDF.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP2.DATA attribute) (nndt.trainable_task.Eikonal3D.DATA attribute) (nndt.trainable_task.SimpleSDF.DATA attribute)

Z

Z (nndt.trainable_task.ApproximateSDF.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP.DATA attribute) (nndt.trainable_task.ApproximateSDFLipMLP2.DATA attribute) (nndt.trainable_task.Eikonal3D.DATA attribute) (nndt.trainable_task.SimpleSDF.DATA attribute)