MCPcopy Create free account

hub / github.com/Relento/lego_release / functions

Functions675 in github.com/Relento/lego_release

↓ 1 callersMethodprepare
(self, predictions, iou_type)
coco_related/coco_eval.py:58
↓ 1 callersMethodprepare_for_coco_detection
(self, predictions)
coco_related/coco_eval.py:68
↓ 1 callersMethodprepare_for_coco_keypoint
(self, predictions)
coco_related/coco_eval.py:127
↓ 1 callersMethodprepare_for_coco_segmentation
(self, predictions)
coco_related/coco_eval.py:92
↓ 1 callersMethodprint_networks
Print the total number of parameters in the network and (if verbose) network architecture Parameters: verbose (bool) -- if verbos
models/base_model.py:198
↓ 1 callersMethodprint_options
Print and save options It will print both current options and default values(if different). It will save options into a text file / [
options/base_options.py:121
↓ 1 callersMethodprint_raw
(self, *args, **kwargs)
tu/loggers/html_table.py:202
↓ 1 callersFunctionrandom_colors
Generate random colors. To get visually distinct colors, generate them in HSV space then convert to RGB.
util/util.py:42
↓ 1 callersFunctionreadFlow
Read .flo file in Middlebury format
tu/flow.py:48
↓ 1 callersMethodreplicate
(self, module, device_ids)
models/heatmap/models/data_parallel.py:74
↓ 1 callersMethodreset
Reset the self.saved status
util/visualizer.py:97
↓ 1 callersFunctionround_corner
Draw a round corner
data_generation/utils.py:20
↓ 1 callersMethodrow
(self, *args, **kwargs)
tu/loggers/html_table.py:140
↓ 1 callersFunctionsafe_json
(data)
tu/configs.py:186
↓ 1 callersMethodsave_figure
(self, figure, row_identifier, col_identifier, ext='png')
tu/loggers/html_table.py:252
↓ 1 callersMethodsave_figure
(self, figure, row_identifier, col_identifier, ext='png')
util/html_table.py:167
↓ 1 callersMethodsave_image
(self, image, row_identifier, col_identifier, ext='png')
tu/loggers/html_table.py:247
↓ 1 callersMethodsave_video
(self, video, row_identifier, col_identifier, ext='mp4')
tu/loggers/html_table.py:257
↓ 1 callersFunctionscatter_map
(obj)
models/heatmap/models/scatter_gather.py:14
↓ 1 callersFunctionsearch_valid_3d_translation
:param bid: :param rot: :param op_type: :param xs: list :param zs: list :param xz_type: 'pos' or 'top_center', showing whethe
lego/utils/inference_utils.py:310
↓ 1 callersFunctionset_axes_equal
Make axes of 3D plot have equal scale so that spheres appear as spheres, cubes as cubes, etc.. This is one possible solution to Matplotlib's
bricks/brick_info.py:852
↓ 1 callersFunctionset_global_definition
(def_)
datasets/definition.py:28
↓ 1 callersFunctionset_seed
(seed)
tu/train_setup.py:8
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process
coco_related/utils.py:256
↓ 1 callersFunctionstring_tuple_to_string
(strings)
tu/configs.py:434
↓ 1 callersMethodtable
(self, name, columns_or_spec_id)
tu/loggers/html_table.py:104
↓ 1 callersMethodtrain
Make models eval mode during test time
models/base_model.py:94
↓ 1 callersFunctionunmold_mask
Converts a mask generated by the neural network to a format similar to its original shape. mask: [height, width] of type float. A small, typic
coco_related/visualize.py:616
↓ 1 callersMethodupdate_candidate
(self, brick: Union[Brick, CBrick], op_type=0)
lego/utils/inference_utils.py:181
↓ 1 callersFunctionvisualize_bricks_pc
(bs: BricksPC, highlight=False, adjust_camera=True)
debug/utils.py:109
↓ 1 callersFunctionyes_or_no
Ask a yes/no question via raw_input() and return their answer. "question" is a string that is presented to the user. "default" is the presumed
util/html_table.py:188
FunctionDataParallel
(module, device_ids=None, output_device=None, dim=0, chunk_sizes=None)
models/heatmap/models/data_parallel.py:120
Method__call__
(self, image, target)
coco_related/coco_utils.py:17
Method__call__
(self, image, target)
coco_related/coco_utils.py:48
Method__call__
(self, image, target)
coco_related/transform.py:20
Method__call__
(self, image, target)
coco_related/transform.py:30
Method__call__
(self, image, target)
coco_related/transform.py:47
Method__getattr__
(self, item)
datasets/definition.py:10
Method__getitem__
(self, y)
tu/configs.py:386
Method__getitem__
(self, index)
datasets/legokps_shape_cond_dataset.py:476
Method__getitem__
Return a data point and its metadata information. Parameters: index - - a random integer for data indexing Returns:
datasets/base_dataset.py:52
Method__getitem__
(self, idx)
coco_related/coco_utils.py:211
Method__init__
(self, opt, val=False)
train_kp.py:76
Method__init__
(self, *args, **kwargs)
tu/configs.py:4
Method__init__
(self, **kwargs)
tu/configs.py:381
Method__init__
:param trigger_period: If None or 0, `check` will always return False.
tu/configs.py:410
Method__init__
(self, data_loader, init_epoch=0)
tu/ddp.py:101
Method__init__
(self, window_size=11, size_average=True, channel=3)
tu/metrics/msssim.py:136
Method__init__
(self, visdir, title, persist_row_counter=False)
tu/loggers/html_table.py:34
Method__init__
(self, opt)
tu/loggers/visualizer.py:15
Method__init__
(self, rasterizer, shader)
data_generation/utils.py:150
Method__init__
Reset the class; indicates the class hasn't been initailized
options/base_options.py:20
Method__init__
(self, bid: str, # List of 3d points. occ_pts: List[np.arra
bricks/brick_info.py:118
Method__init__
(self, bid, l, w, h, is_base=False)
bricks/brick_info.py:176
Method__init__
(self, bid, l, w)
bricks/brick_info.py:193
Method__init__
(self, bid, l, w, h, stud_pts, rotations, occ_pts=None)
bricks/brick_info.py:202
Method__init__
Args: n: number of bricks to be vertically stacked brick_type: position: rotation:
bricks/brick_info.py:933
Method__init__
(self, b1_brick_type, b1_position, b1_rotation, b2_brick_type='3024')
bricks/brick_info.py:1029
Method__init__
(self, grid_size=(21, 21, 21), record_parents=True)
bricks/brick_info.py:1168
Method__init__
(self, bricks_pc: BricksPC, position, rotation)
bricks/brick_info.py:1550
Method__init__
(self, opt, val)
datasets/legokps_shape_cond_dataset.py:180
Method__init__
Initialize the class; save the options in the class Parameters: opt (Option class)-- stores all the experiment flags; needs to be
datasets/base_dataset.py:24
Method__init__
Initialize this class Step 1: create a dataset instance given the name [dataset_mode] Step 2: create a multi-threaded data loader.
datasets/__init__.py:73
Method__init__
(self, base_optimizer, nr_acc)
models/utils.py:7
Method__init__
(self)
models/utils.py:148
Method__init__
Initialize this model class. Parameters: opt -- training/test options A few things can be done here. - (required
models/hourglass_shape_cond_model.py:106
Method__init__
Initialize the BaseModel class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseO
models/base_model.py:20
Method__init__
Initialize this model class. Parameters: opt -- training/test options A few things can be done here. - (required
models/hourglass_trans_model.py:105
Method__init__
(self, pretrain=False, use_layer2=False)
models/networks.py:160
Method__init__
(self, pretrain=False, use_layer2=False)
models/networks.py:177
Method__init__
(self, pretrain=False)
models/networks.py:197
Method__init__
(self)
models/coordconv.py:6
Method__init__
(self, opt)
models/heatmap/loss.py:74
Method__init__
(self, opt, num_classes, occ_out_channels, occ_fmap_size, shape_condition=False, brick_emb=64
models/heatmap/lego_hg.py:28
Method__init__
(self, opt, num_classes, occ_out_channels, occ_fmap_size, brick_emb=64, num_
models/heatmap/lego_hg.py:204
Method__init__
:param stack_4rots: stacking occ for four different rotations
models/heatmap/lego_hg.py:326
Method__init__
(self)
models/heatmap/lego_hg.py:435
Method__init__
(self, num_features=64, feature_map_size=(32, 32))
models/heatmap/lego_hg.py:443
Method__init__
(self, num_embedding, embedding_dim)
models/heatmap/lego_hg.py:463
Method__init__
(self)
models/heatmap/models/losses.py:116
Method__init__
(self)
models/heatmap/models/losses.py:133
Method__init__
(self)
models/heatmap/models/losses.py:143
Method__init__
(self)
models/heatmap/models/losses.py:156
Method__init__
(self)
models/heatmap/models/losses.py:171
Method__init__
(self)
models/heatmap/models/losses.py:195
Method__init__
(self, opt)
models/heatmap/models/losses.py:207
Method__init__
(self)
models/heatmap/models/losses.py:249
Method__init__
(self, module, device_ids=None, output_device=None, dim=0, chunk_sizes=None)
models/heatmap/models/data_parallel.py:44
Method__init__
(self, k, inp_dim, out_dim, stride=1, with_bn=True)
models/heatmap/models/networks/hourglass.py:10
Method__init__
(self, inp_dim, out_dim, with_bn=True)
models/heatmap/models/networks/hourglass.py:26
Method__init__
(self, k, inp_dim, out_dim, stride=1, with_bn=True)
models/heatmap/models/networks/hourglass.py:43
Method__init__
( self, n, dims, modules, layer=residual, make_up_layer=make_layer, make_low_layer=mak
models/heatmap/models/networks/hourglass.py:122
Method__init__
(self, heads, num_stacks=2, img_nc=3, cond_emb_dim=0, downsample_2x=False)
models/heatmap/models/networks/hourglass.py:314
Method__init__
(self, visdir, title)
util/html_table.py:36
Method__init__
(self, technique='cyclegan', verbose=True)
util/get_data.py:29
Method__init__
Initialize the HTML classes Parameters: web_dir (str) -- a directory that stores the webpage. HTML file will be created at <web_d
util/html.py:15
Method__init__
Initialize the Visualizer class Parameters: opt -- stores all the experiment flags; needs to be a subclass of BaseOptions
util/visualizer.py:62
Method__init__
Initialize the ImagePool class Parameters: pool_size (int) -- the size of image buffer, if pool_size=0, no buffer will be created
util/image_pool.py:13
Method__init__
For conveniently render the scenario where we have a base bricks_pc and want to render multiple candidate bricks. :param bric
lego/utils/inference_utils.py:144
Method__init__
(self, window_size=20, fmt=None)
coco_related/utils.py:17
← previousnext →301–400 of 675, ranked by callers