MCPcopy Create free account

hub / github.com/magicleap/Atlas / functions

Functions135 in github.com/magicleap/Atlas

⨍ Functions 135◇ Types & classes 29
Method__init__
(self, hparams)
atlas/model.py:89
Method__init__
(self, n, norm='BN', condition=True)
atlas/backbone3d.py:110
Method__init__
(self, channels=[32,64,128], layers_down=[1,2,3], layers_up=[3,3,3], norm='BN', drop=0, zero_
atlas/backbone3d.py:137
Method__len__
(self)
atlas/data.py:120
Method__len__
(self)
atlas/data.py:196
Method__repr__
(self)
atlas/transforms.py:107
Method__repr__
(self)
atlas/transforms.py:190
Method__repr__
(self)
atlas/transforms.py:272
Method__repr__
(self)
atlas/transforms.py:288
Functioncollate_fn
Flatten a set of items from ScenesDataset into a batch. Pytorch dataloader has memory issues with nested and complex data structures. This
atlas/data.py:233
Methodconfigure_optimizers
(self)
atlas/model.py:414
Methoddelete
(self)
evaluate.py:69
Methodexperiment1
(self)
atlas/logger.py:50
Methodforward
(self, features)
atlas/backbone2d.py:71
Methodforward
(self, x, targets=None)
atlas/heads3d.py:46
Methodforward
(self, xs, targets=None)
atlas/heads3d.py:89
Methodforward
(self, xs, targets=None)
atlas/heads3d.py:183
Methodforward
(self, xs, targets=None)
atlas/heads3d.py:236
Methodforward
(self, x, targets=None)
atlas/heads2d.py:37
Methodforward
(self, x, targets=None)
atlas/heads2d.py:63
Methodforward
(self, x)
atlas/backbone3d.py:77
Methodforward
Args: x: tensor from encoder y: tensor from decoder mask
atlas/backbone3d.py:118
Methodforward
(self, x)
atlas/backbone3d.py:183
Methodget_mesh
(self)
atlas/data.py:163
Methodon_train_start
(self, trainer, pl_module)
train.py:30
Methodon_validation_end
(self, trainer, pl_module)
train.py:34
Methodon_validation_start
(self, trainer, pl_module)
train.py:32
Functionprepare_rio_scene
(scene, path, path_meta, verbose=2)
atlas/datasets/rio.py:74
Functionprepare_rio_splits
Generates txt files for each dataset split Creates a folder 'data' in the current working directory. In that folder we create a split.txt fi
atlas/datasets/rio.py:140
Methodreset
Initialize the volumes to default values
atlas/tsdf.py:359
Methodtrain_dataloader
(self)
atlas/model.py:345
Methodtraining_step
(self, batch, batch_idx)
atlas/model.py:370
Methodval_dataloader
(self)
atlas/model.py:357
Methodvalidation_epoch_end
(self, outputs)
atlas/model.py:407
Methodvalidation_step
(self, batch, batch_idx)
atlas/model.py:394
← previous101–135 of 135, ranked by callers