MCPcopy Create free account

hub / github.com/Relento/lego_release / types & classes

Types & classes88 in github.com/Relento/lego_release

↓ 47 callersClassSimpleBrickInfo
Define simple rectangular bricks by its size
bricks/brick_info.py:171
↓ 19 callersClassHTMLTableColumnDesc
util/html_table.py:13
↓ 13 callersClassCBrick
bricks/brick_info.py:1547
↓ 11 callersClassBrick
bricks/brick_info.py:908
↓ 11 callersClassSlopeBrickInfo
bricks/brick_info.py:201
↓ 9 callersClassTileBrickInfo
bricks/brick_info.py:192
↓ 5 callersClassAddCoords3D
models/coordconv.py:5
↓ 5 callersClassBasicBrickInfo
bricks/brick_info.py:117
↓ 5 callersClassMeters
models/utils.py:147
↓ 4 callersClassBricksPC
bricks/brick_info.py:1167
↓ 3 callersClassVoxelTransformEncoder
models/heatmap/lego_hg.py:325
↓ 3 callersClassconvolution
models/heatmap/models/networks/hourglass.py:9
↓ 2 callersClassAccumGrad
models/utils.py:6
↓ 2 callersClassAttrDict
Dict as attribute trick.
tu/configs.py:1
↓ 2 callersClassAvgPool
models/heatmap/lego_hg.py:426
↓ 2 callersClassBrickVoxelProjectionEncoder
models/heatmap/lego_hg.py:442
↓ 2 callersClassFlatten
models/heatmap/lego_hg.py:434
↓ 2 callersClassHTMLTableColumnDesc
tu/loggers/html_table.py:13
↓ 2 callersClassHTMLTableVisualizer
Example: >>> vis = HTMLTableVisualizer('<some_dir>', 'Visualization') >>> with vis.html(): >>> with vis.table('Table
tu/loggers/html_table.py:20
↓ 2 callersClassL1Loss
models/heatmap/models/losses.py:183
↓ 2 callersClassLegoHourGlass
Main class for Generalized R-CNN. Arguments: backbone (nn.Module): rpn (nn.Module): roi_heads (nn.Module): takes the
models/heatmap/lego_hg.py:15
↓ 2 callersClassMaskLoss
models/heatmap/loss.py:73
↓ 2 callersClassSimpleShader
data_generation/utils.py:40
↓ 2 callersClassSmoothedValue
Track a series of values and provide access to smoothed values over a window or the global series average.
coco_related/utils.py:12
↓ 2 callersClasskp_module
models/heatmap/models/networks/hourglass.py:121
↓ 2 callersClassresidual
models/heatmap/models/networks/hourglass.py:42
↓ 1 callersClassBaseHTMLHelper
Handles higher level logic for HTMLTableVisualizer
tu/loggers/html_helper.py:9
↓ 1 callersClassCocoDetection
coco_related/coco_utils.py:206
↓ 1 callersClassConvertCocoPolysToMask
coco_related/coco_utils.py:47
↓ 1 callersClassCustomDatasetDataLoader
Wrapper class of Dataset class that performs multi-threaded data loading
datasets/__init__.py:70
↓ 1 callersClassDistributedCustomDatasetDataloader
train_kp.py:75
↓ 1 callersClassDistributedTrainOptions
This class includes training options. It also includes shared options defined in BaseOptions.
train_kp.py:99
↓ 1 callersClassEvalOptions
eval.py:301
↓ 1 callersClassFocalLoss
nn.Module warpper for focal loss
models/heatmap/models/losses.py:113
↓ 1 callersClassGlobalDefinitionWrapper
datasets/definition.py:9
↓ 1 callersClassHBrick
bricks/brick_info.py:1026
↓ 1 callersClassHTML
This HTML class allows us to save images and write texts into a single HTML file. It consists of functions such as <add_header> (add a text head
util/html.py:7
↓ 1 callersClassHTMLTableVisualizer
Example: >>> vis = HTMLTableVisualizer('<some_dir>', 'Visualization') >>> with vis.html(): >>> with vis.table('Table
util/html_table.py:22
↓ 1 callersClassHourglassNet
models/heatmap/models/networks/hourglass.py:313
↓ 1 callersClassIdentity
models/networks.py:14
↓ 1 callersClassKPLoss
models/heatmap/loss.py:11
↓ 1 callersClassLegoKPSDefinition
datasets/legokps_shape_cond_dataset.py:824
↓ 1 callersClassMergeUp
models/heatmap/models/networks/hourglass.py:86
↓ 1 callersClassMeshRendererWithDepth
data_generation/utils.py:149
↓ 1 callersClassRenderBricks
lego/utils/inference_utils.py:140
↓ 1 callersClassRotLoss
models/heatmap/models/losses.py:194
↓ 1 callersClassTagLoss
models/heatmap/models/losses.py:206
↓ 1 callersClassTestOptions
This class includes test options. It also includes shared options defined in BaseOptions.
options/test_options.py:4
↓ 1 callersClassTrainOptions
This class includes training options. It also includes shared options defined in BaseOptions.
options/train_options.py:4
↓ 1 callersClassVBrick
bricks/brick_info.py:930
↓ 1 callersClassVisualizer
This class includes several functions that can display/save images and print/save logging information. It uses a Python library 'visdom' for disp
util/visualizer.py:56
↓ 1 callersClassVoxelEmbedding
models/heatmap/lego_hg.py:462
↓ 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
models/heatmap/models/data_parallel.py:10
↓ 1 callersClass_Logger
tu/configs.py:444
ClassBaseDataset
This class is an abstract base class (ABC) for datasets. To create a subclass, you need to implement the following four functions: -- <__init
datasets/base_dataset.py:14
ClassBaseModel
This class is an abstract base class (ABC) for models. To create a subclass, you need to implement the following five functions: -- <__ini
models/base_model.py:10
ClassBaseOptions
This class defines options used during both training and test time. It also implements several helper functions such as parsing, printing, and sa
options/base_options.py:13
ClassBaseTestCase
tu/test_utils.py:13
ClassBinRotLoss
models/heatmap/models/losses.py:248
ClassCocoEvaluator
coco_related/coco_eval.py:15
ClassCompose
coco_related/transform.py:16
ClassConditionTimer
A timer that goes off after the a fixed time period. The catch: you need to poll it and provide it the time! Usage: ``` timer =
tu/configs.py:395
ClassDatasetDefinitionBase
datasets/definition.py:4
ClassFilterAndRemapCocoCategories
coco_related/coco_utils.py:12
ClassGetData
A Python script for downloading CycleGAN or pix2pix datasets. Parameters: technique (str) -- One of: 'cyclegan' or 'pix2pix'. ver
util/get_data.py:13
ClassHTMLVisualizer
tu/loggers/visualizer.py:14
ClassHourglassShapeCondModel
models/hourglass_shape_cond_model.py:40
ClassHourglassTransModel
models/hourglass_trans_model.py:38
ClassImagePool
This class implements an image buffer that stores previously generated images. This buffer enables us to update discriminators using a history of
util/image_pool.py:6
ClassInfDataLoader
tu/ddp.py:100
ClassIntIdDict
Automatically assign int IDs to hashable objects. Usage: ``` id_map = IntIdDict() print(id_map['a']) print(id_map['b'])
tu/configs.py:344
ClassLegoHourGlassMask
models/heatmap/lego_hg.py:203
ClassLegoKpsShapeCondDataset
datasets/legokps_shape_cond_dataset.py:140
ClassMSSSIM
tu/metrics/msssim.py:135
ClassMetricLogger
coco_related/utils.py:144
ClassNormRegL1Loss
models/heatmap/models/losses.py:155
ClassRandomHorizontalFlip
coco_related/transform.py:26
ClassRegL1Loss
models/heatmap/models/losses.py:142
ClassRegLoss
Regression loss for an output tensor Arguments: output (batch x dim x h x w) mask (batch x max_objects) ind (batch x max
models/heatmap/models/losses.py:124
ClassRegWeightedL1Loss
models/heatmap/models/losses.py:170
ClassSSIM
tu/metrics/msssim.py:111
ClassSimpleResNetEncoder
models/networks.py:141
ClassSimpleResNetEncoder2
models/networks.py:159
ClassSimpleResNetEncoder3
models/networks.py:176
ClassSimpleResNetEncoder4
models/networks.py:196
ClassToTensor
coco_related/transform.py:46
Classexkp
models/heatmap/models/networks/hourglass.py:196
Classfully_connected
models/heatmap/models/networks/hourglass.py:25