MCPcopy Create free account

hub / github.com/ZhangYushan3/GMSF / types & classes

Types & classes21 in github.com/ZhangYushan3/GMSF

⨍ Functions 170◇ Types & classes 21
↓ 4 callersClassFlyingThings3D_subset
dataloader/flyingthings3d.py:10
↓ 2 callersClassCropGroundAuto
dataloader/waymo_utils/crop_ground.py:4
↓ 2 callersClassFlyingThings3D_flownet3d
dataloader/flyingthings3d.py:82
↓ 2 callersClassGMSF
models/gmsf.py:9
↓ 2 callersClassKITTI_hplflownet
Args: train (bool): If True, creates dataset from training set, otherwise creates from test set. transform (callable): ge
dataloader/kitti.py:10
↓ 2 callersClassTransformerLayer
models/transformer.py:17
↓ 2 callersClassWaymo
dataloader/waymo.py:12
↓ 1 callersClassDGCNN
models/backbone.py:70
↓ 1 callersClassFeatureTransformer3D
models/transformer.py:110
↓ 1 callersClassFeatureTransformer3D_PT
models/transformer.py:242
↓ 1 callersClassKITTI_flownet3d
dataloader/kitti.py:120
↓ 1 callersClassLogger
utils/logger.py:12
↓ 1 callersClassMLP
models/backbone.py:122
↓ 1 callersClassPointNet
models/backbone.py:46
↓ 1 callersClassPreprocessor
utils/preprocess_flyingthings3d_subset.py:57
↓ 1 callersClassProcessData
dataloader/transforms.py:23
↓ 1 callersClassSelfCorrelationSoftmax3D
flow propagation with self-attention on feature query: feature0, key: feature0, value: flow
models/matching.py:48
↓ 1 callersClassTransformerBlock
self attention + cross attention + FFN
models/transformer.py:72
↓ 1 callersClassTransformerBlock_PT
models/transformer.py:200
↓ 1 callersClassWaymo2KITTI
Waymo to KITTI converter. This class serves as the converter to change the waymo raw data to KITTI format. Args: load_dir (str):
waymo_tools/waymo_converter.py:18
ClassWaymoSFDataset
Waymo scene flow dataset. TODO: Add online mode to process data during training to save storage space.
dataloader/waymo_sf_dataset.py:14