Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Relento/lego_release
/ types & classes
Types & classes
88 in github.com/Relento/lego_release
⨍
Functions
675
◇
Types & classes
88
↓ 47 callers
Class
SimpleBrickInfo
Define simple rectangular bricks by its size
bricks/brick_info.py:171
↓ 19 callers
Class
HTMLTableColumnDesc
util/html_table.py:13
↓ 13 callers
Class
CBrick
bricks/brick_info.py:1547
↓ 11 callers
Class
Brick
bricks/brick_info.py:908
↓ 11 callers
Class
SlopeBrickInfo
bricks/brick_info.py:201
↓ 9 callers
Class
TileBrickInfo
bricks/brick_info.py:192
↓ 5 callers
Class
AddCoords3D
models/coordconv.py:5
↓ 5 callers
Class
BasicBrickInfo
bricks/brick_info.py:117
↓ 5 callers
Class
Meters
models/utils.py:147
↓ 4 callers
Class
BricksPC
bricks/brick_info.py:1167
↓ 3 callers
Class
VoxelTransformEncoder
models/heatmap/lego_hg.py:325
↓ 3 callers
Class
convolution
models/heatmap/models/networks/hourglass.py:9
↓ 2 callers
Class
AccumGrad
models/utils.py:6
↓ 2 callers
Class
AttrDict
Dict as attribute trick.
tu/configs.py:1
↓ 2 callers
Class
AvgPool
models/heatmap/lego_hg.py:426
↓ 2 callers
Class
BrickVoxelProjectionEncoder
models/heatmap/lego_hg.py:442
↓ 2 callers
Class
Flatten
models/heatmap/lego_hg.py:434
↓ 2 callers
Class
HTMLTableColumnDesc
tu/loggers/html_table.py:13
↓ 2 callers
Class
HTMLTableVisualizer
Example: >>> vis = HTMLTableVisualizer('<some_dir>', 'Visualization') >>> with vis.html(): >>> with vis.table('Table
tu/loggers/html_table.py:20
↓ 2 callers
Class
L1Loss
models/heatmap/models/losses.py:183
↓ 2 callers
Class
LegoHourGlass
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 callers
Class
MaskLoss
models/heatmap/loss.py:73
↓ 2 callers
Class
SimpleShader
data_generation/utils.py:40
↓ 2 callers
Class
SmoothedValue
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 callers
Class
kp_module
models/heatmap/models/networks/hourglass.py:121
↓ 2 callers
Class
residual
models/heatmap/models/networks/hourglass.py:42
↓ 1 callers
Class
BaseHTMLHelper
Handles higher level logic for HTMLTableVisualizer
tu/loggers/html_helper.py:9
↓ 1 callers
Class
CocoDetection
coco_related/coco_utils.py:206
↓ 1 callers
Class
ConvertCocoPolysToMask
coco_related/coco_utils.py:47
↓ 1 callers
Class
CustomDatasetDataLoader
Wrapper class of Dataset class that performs multi-threaded data loading
datasets/__init__.py:70
↓ 1 callers
Class
DistributedCustomDatasetDataloader
train_kp.py:75
↓ 1 callers
Class
DistributedTrainOptions
This class includes training options. It also includes shared options defined in BaseOptions.
train_kp.py:99
↓ 1 callers
Class
EvalOptions
eval.py:301
↓ 1 callers
Class
FocalLoss
nn.Module warpper for focal loss
models/heatmap/models/losses.py:113
↓ 1 callers
Class
GlobalDefinitionWrapper
datasets/definition.py:9
↓ 1 callers
Class
HBrick
bricks/brick_info.py:1026
↓ 1 callers
Class
HTML
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 callers
Class
HTMLTableVisualizer
Example: >>> vis = HTMLTableVisualizer('<some_dir>', 'Visualization') >>> with vis.html(): >>> with vis.table('Table
util/html_table.py:22
↓ 1 callers
Class
HourglassNet
models/heatmap/models/networks/hourglass.py:313
↓ 1 callers
Class
Identity
models/networks.py:14
↓ 1 callers
Class
KPLoss
models/heatmap/loss.py:11
↓ 1 callers
Class
LegoKPSDefinition
datasets/legokps_shape_cond_dataset.py:824
↓ 1 callers
Class
MergeUp
models/heatmap/models/networks/hourglass.py:86
↓ 1 callers
Class
MeshRendererWithDepth
data_generation/utils.py:149
↓ 1 callers
Class
RenderBricks
lego/utils/inference_utils.py:140
↓ 1 callers
Class
RotLoss
models/heatmap/models/losses.py:194
↓ 1 callers
Class
TagLoss
models/heatmap/models/losses.py:206
↓ 1 callers
Class
TestOptions
This class includes test options. It also includes shared options defined in BaseOptions.
options/test_options.py:4
↓ 1 callers
Class
TrainOptions
This class includes training options. It also includes shared options defined in BaseOptions.
options/train_options.py:4
↓ 1 callers
Class
VBrick
bricks/brick_info.py:930
↓ 1 callers
Class
Visualizer
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 callers
Class
VoxelEmbedding
models/heatmap/lego_hg.py:462
↓ 1 callers
Class
_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 callers
Class
_Logger
tu/configs.py:444
Class
BaseDataset
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
Class
BaseModel
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
Class
BaseOptions
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
Class
BaseTestCase
tu/test_utils.py:13
Class
BinRotLoss
models/heatmap/models/losses.py:248
Class
CocoEvaluator
coco_related/coco_eval.py:15
Class
Compose
coco_related/transform.py:16
Class
ConditionTimer
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
Class
DatasetDefinitionBase
datasets/definition.py:4
Class
FilterAndRemapCocoCategories
coco_related/coco_utils.py:12
Class
GetData
A Python script for downloading CycleGAN or pix2pix datasets. Parameters: technique (str) -- One of: 'cyclegan' or 'pix2pix'. ver
util/get_data.py:13
Class
HTMLVisualizer
tu/loggers/visualizer.py:14
Class
HourglassShapeCondModel
models/hourglass_shape_cond_model.py:40
Class
HourglassTransModel
models/hourglass_trans_model.py:38
Class
ImagePool
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
Class
InfDataLoader
tu/ddp.py:100
Class
IntIdDict
Automatically assign int IDs to hashable objects. Usage: ``` id_map = IntIdDict() print(id_map['a']) print(id_map['b'])
tu/configs.py:344
Class
LegoHourGlassMask
models/heatmap/lego_hg.py:203
Class
LegoKpsShapeCondDataset
datasets/legokps_shape_cond_dataset.py:140
Class
MSSSIM
tu/metrics/msssim.py:135
Class
MetricLogger
coco_related/utils.py:144
Class
NormRegL1Loss
models/heatmap/models/losses.py:155
Class
RandomHorizontalFlip
coco_related/transform.py:26
Class
RegL1Loss
models/heatmap/models/losses.py:142
Class
RegLoss
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
Class
RegWeightedL1Loss
models/heatmap/models/losses.py:170
Class
SSIM
tu/metrics/msssim.py:111
Class
SimpleResNetEncoder
models/networks.py:141
Class
SimpleResNetEncoder2
models/networks.py:159
Class
SimpleResNetEncoder3
models/networks.py:176
Class
SimpleResNetEncoder4
models/networks.py:196
Class
ToTensor
coco_related/transform.py:46
Class
exkp
models/heatmap/models/networks/hourglass.py:196
Class
fully_connected
models/heatmap/models/networks/hourglass.py:25