Index

A

aidsorb module aidsorb.data module aidsorb.datamodules module aidsorb.litmodels module aidsorb.models module

aidsorb.modules module aidsorb.transforms module aidsorb.utils module aidsorb.visualize module

C

Center (class in aidsorb.transforms) Collator (class in aidsorb.data)

configure_optimizers() (aidsorb.litmodels.PointLit method) conv1d_block() (in module aidsorb.modules)

D

dense_block() (in module aidsorb.modules)

draw_pcd() (in module aidsorb.visualize) draw_pcd_from_file() (in module aidsorb.visualize)

F

forward() (aidsorb.litmodels.PointLit method) (aidsorb.models.PointNet method) (aidsorb.modules.PointNetBackbone method) (aidsorb.modules.PointNetClsHead method) (aidsorb.modules.PointNetSegHead method) (aidsorb.modules.TNet method)

G

get_atom_colors() (in module aidsorb.visualize)

get_elements() (in module aidsorb.visualize) get_names() (in module aidsorb.data)

I

Identity (class in aidsorb.transforms)

J

Jitter (class in aidsorb.transforms)

M

module aidsorb aidsorb.data aidsorb.datamodules aidsorb.litmodels aidsorb.models aidsorb.modules aidsorb.transforms aidsorb.utils aidsorb.visualize

P

pad_pcds() (in module aidsorb.data) pcd_from_dir() (in module aidsorb.utils) pcd_from_file() (in module aidsorb.utils) pcd_from_files() (in module aidsorb.utils) pcd_names (aidsorb.data.PCDDataset property) PCDDataModule (class in aidsorb.datamodules) PCDDataset (class in aidsorb.data)

PointLit (class in aidsorb.litmodels) PointNet (class in aidsorb.models) PointNetBackbone (class in aidsorb.modules) PointNetClsHead (class in aidsorb.modules) PointNetSegHead (class in aidsorb.modules) predict_step() (aidsorb.litmodels.PointLit method) prepare_data() (in module aidsorb.data)

R

RandomErase (class in aidsorb.transforms)

RandomRotation (class in aidsorb.transforms)

S

set_test_dataset() (aidsorb.datamodules.PCDDataModule method) set_train_dataset() (aidsorb.datamodules.PCDDataModule method)

set_validation_dataset() (aidsorb.datamodules.PCDDataModule method) setup() (aidsorb.datamodules.PCDDataModule method) split_pcd() (in module aidsorb.utils)

T

test_dataloader() (aidsorb.datamodules.PCDDataModule method) test_names (aidsorb.datamodules.PCDDataModule property) test_step() (aidsorb.litmodels.PointLit method) TNet (class in aidsorb.modules)

train_dataloader() (aidsorb.datamodules.PCDDataModule method) train_names (aidsorb.datamodules.PCDDataModule property) training_step() (aidsorb.litmodels.PointLit method) transform_pcd() (in module aidsorb.transforms)

U

upsample_pcd() (in module aidsorb.data)

V

val_dataloader() (aidsorb.datamodules.PCDDataModule method)

val_names (aidsorb.datamodules.PCDDataModule property) validation_step() (aidsorb.litmodels.PointLit method)