Types & classes — github.com/NVlabs/CenterPose

CodeHub MCPcopy Create free account

hub / github.com/NVlabs/CenterPose / types & classes

Types & classes95 in github.com/NVlabs/CenterPose

⨍ Functions 615◇ Types & classes 95

↓ 21 callersClassopts

src/lib/opts.py:14

↓ 10 callersClassDLA

src/lib/models/networks/dlav0.py:222

↓ 8 callersClassDCNv2Pooling

src/lib/models/networks/DCNv2/dcn_v2.py:187

↓ 6 callersClassTree

src/lib/models/networks/dlav0.py:166

↓ 6 callersClassTree

src/lib/models/networks/pose_dla_dcn.py:171

↓ 5 callersClassPoint

data/bbox_collision_detection.py:15

↓ 4 callersClassCheckLineBox

data/bbox_collision_detection.py:23

↓ 4 callersClassDCN

src/lib/models/networks/DCNv2/dcn_v2.py:97

↓ 3 callersClassAverageMeter

Computes and stores the average and current value

src/lib/utils/utils.py:6

↓ 3 callersClassConvGRU

src/lib/models/networks/convGRU.py:54

↓ 3 callersClassLogger

src/lib/logger.py:24

↓ 3 callersClassconvolution

src/lib/models/networks/large_hourglass.py:18

↓ 2 callersClassDCNPooling

src/lib/models/networks/DCNv2/dcn_v2.py:223

↓ 2 callersClassDCNv2

src/lib/models/networks/DCNv2/dcn_v2.py:57

↓ 2 callersClassDLASeg

src/lib/models/networks/pose_dla_dcn.py:457

↓ 2 callersClassDebugger

src/lib/utils/debugger.py:10

↓ 2 callersClassDeformConv

src/lib/models/networks/pose_dla_dcn.py:377

↓ 2 callersClassEvaluator

Class for evaluating a deep pursuit model.

src/tools/objectron_eval/eval_video_official.py:159

↓ 2 callersClassFocalLoss

nn.Module warpper for focal loss

src/lib/models/losses.py:127

↓ 2 callersClassIDAUp

src/lib/models/networks/pose_dla_dcn.py:392

↓ 2 callersClassIdentity

src/lib/models/networks/dlav0.py:449

↓ 2 callersClasseval_opts

src/tools/objectron_eval/eval_opts.py:12

↓ 2 callersClasskp_module

src/lib/models/networks/large_hourglass.py:130

↓ 2 callersClassresidual

src/lib/models/networks/large_hourglass.py:51

↓ 1 callersClassConvGRUCell

src/lib/models/networks/convGRU.py:7

↓ 1 callersClassCuboid3d

This class contains a 3D cuboid.

src/lib/utils/pnp/cuboid_objectron.py:49

↓ 1 callersClassCuboidPNPSolver

This class is used to find the 6-DoF pose of a cuboid given its projected vertices. Runs perspective-n-point (PNP) algorithm.

src/lib/utils/pnp/cuboid_pnp_solver.py:13

↓ 1 callersClassDLA

src/lib/models/networks/pose_dla_dcn.py:227

↓ 1 callersClassDLASeg

src/lib/models/networks/dlav0.py:564

↓ 1 callersClassDLAUp

src/lib/models/networks/dlav0.py:528

↓ 1 callersClassDLAUp

src/lib/models/networks/pose_dla_dcn.py:420

↓ 1 callersClassEvaluator

Class for evaluating a deep pursuit model.

data/objectron/dataset/eval.py:47

↓ 1 callersClassEvaluator

Class for evaluating a deep pursuit model.

src/tools/objectron_eval/group_report_new.py:48

↓ 1 callersClassEvaluator

Class for evaluating a deep pursuit model.

src/tools/objectron_eval/eval_image_official.py:129

↓ 1 callersClassEvaluator

Class for evaluating a deep pursuit model.

src/tools/objectron_eval/group_report_old.py:45

↓ 1 callersClassEvaluator

Class for evaluating a deep pursuit model.

src/tools/objectron_eval/objectron/dataset/eval.py:49

↓ 1 callersClassHourglassNet

src/lib/models/networks/large_hourglass.py:296

↓ 1 callersClassIDAUp

src/lib/models/networks/dlav0.py:469

↓ 1 callersClassMergeUp

src/lib/models/networks/large_hourglass.py:95

↓ 1 callersClassModelWithLoss

src/lib/trains/base_trainer.py:17

↓ 1 callersClassObjectPoseLoss

src/lib/trains/object_pose.py:22

↓ 1 callersClassObjectronreader

src/tools/objectron_eval/prepare_test_video.py:32

↓ 1 callersClassPoseResNet

src/lib/models/networks/msra_resnet.py:108

↓ 1 callersClassPoseResNet

src/lib/models/networks/resnet_dcn.py:134

↓ 1 callersClassRegKLDKeyLoss

src/lib/models/losses.py:208

↓ 1 callersClassRegKLDScaleLoss

src/lib/models/losses.py:183

↓ 1 callersClassRegL1Loss

src/lib/models/losses.py:156

↓ 1 callersClassRegLoss

Regression loss for an output tensor Arguments: output (batch x dim x h x w) mask (batch x max_objects) ind (batch x max

src/lib/models/losses.py:138

↓ 1 callersClassRegWeightedL1Loss

src/lib/models/losses.py:245

↓ 1 callersClassRoot

src/lib/models/networks/dlav0.py:145

↓ 1 callersClassRoot

src/lib/models/networks/pose_dla_dcn.py:150

↓ 1 callersClassStruct

src/lib/opts.py:494

↓ 1 callersClassTracker

src/lib/utils/tracker.py:14

↓ 1 callersClassTracker_baseline

src/lib/utils/tracker_baseline.py:14

↓ 1 callersClass_DataParallel

r"""Implements data parallelism at the module level. This container parallelizes the application of the given module by splitting the input a

src/lib/models/data_parallel.py:10

Class for accuracy metric.

data/objectron/dataset/metrics.py:101

Class for accuracy metric.

src/tools/objectron_eval/objectron/dataset/metrics.py:101

Class for accuracy metric.

src/tools/objectron_eval/objectron/dataset/metrics_nvidia.py:140

ClassAveragePrecision

Class for computing average precision.

data/objectron/dataset/metrics.py:31

ClassAveragePrecision

Class for computing average precision.

src/tools/objectron_eval/objectron/dataset/metrics.py:31

ClassAveragePrecision

Class for computing average precision.

src/tools/objectron_eval/objectron/dataset/metrics_nvidia.py:46

ClassBaseDetector

src/lib/detectors/base_detector.py:31

ClassBaseTrainer

src/lib/trains/base_trainer.py:34

ClassBasicBlock

src/lib/models/networks/dlav0.py:29

ClassBasicBlock

src/lib/models/networks/msra_resnet.py:35

ClassBasicBlock

src/lib/models/networks/resnet_dcn.py:39

ClassBasicBlock

src/lib/models/networks/pose_dla_dcn.py:34

ClassBinRotLoss

src/lib/models/losses.py:269

ClassBottleneck

src/lib/models/networks/dlav0.py:60

ClassBottleneck

src/lib/models/networks/msra_resnet.py:67

ClassBottleneck

src/lib/models/networks/resnet_dcn.py:71

ClassBottleneck

src/lib/models/networks/pose_dla_dcn.py:65

ClassBottleneckX

src/lib/models/networks/dlav0.py:101

ClassBottleneckX

src/lib/models/networks/pose_dla_dcn.py:106

General 3D Oriented Bounding Box.

data/objectron/dataset/box.py:44

General 3D Oriented Bounding Box.

src/tools/objectron_eval/objectron/dataset/box.py:48

ClassCuboidVertexType

src/lib/utils/pnp/cuboid_objectron.py:13

Class for recording hits and misses of detection results.

data/objectron/dataset/metrics.py:6

Class for recording hits and misses of detection results.

src/tools/objectron_eval/objectron/dataset/metrics.py:6

Class for recording hits and misses of detection results.

src/tools/objectron_eval/objectron/dataset/metrics_nvidia.py:13

src/lib/models/networks/pose_dla_dcn.py:349

ClassInterpolate

src/lib/models/networks/pose_dla_dcn.py:446

General Intersection Over Union cost for Oriented 3D bounding boxes.

data/objectron/dataset/iou.py:14

General Intersection Over Union cost for Oriented 3D bounding boxes.

src/tools/objectron_eval/objectron/dataset/iou.py:14

src/lib/models/losses.py:258

ClassNormRegL1Loss

src/lib/models/losses.py:230

ClassObjectPoseDataset

src/lib/datasets/dataset_combined.py:54

ClassObjectPoseDetector

src/lib/detectors/object_pose.py:126

ClassObjectPoseTrainer

src/lib/trains/object_pose.py:208

ClassObjectronParser

Parser using NumPy.

data/objectron/dataset/parser.py:15

ClassObjectronParser

Parser using NumPy.

src/tools/objectron_eval/objectron/dataset/parser.py:57

src/lib/models/networks/DCNv2/dcn_v2.py:16

Class_DCNv2Pooling

src/lib/models/networks/DCNv2/dcn_v2.py:132

src/lib/models/networks/large_hourglass.py:190

Classfully_connected

src/lib/models/networks/large_hourglass.py:34