MCPcopy Create free account

hub / github.com/Relento/lego_release / functions

Functions675 in github.com/Relento/lego_release

⨍ Functions 675◇ Types & classes 88
↓ 106 callersFunctionprint
(*args, **kwargs)
coco_related/utils.py:263
↓ 83 callersMethodto
(self, device)
data_generation/utils.py:155
↓ 45 callersMethod_print
(self, *args, **kwargs)
tu/loggers/html_table.py:198
↓ 38 callersMethodmax
(self)
coco_related/utils.py:58
↓ 29 callersMethod_print
(self, *args, **kwargs)
util/html_table.py:147
↓ 28 callersFunction_transpose_and_gather_feat
(feat, ind)
models/heatmap/models/utils.py:24
↓ 27 callersFunction_gather_feat
(feat, ind, mask=None)
models/heatmap/models/utils.py:13
↓ 25 callersMethodupdate
(self, name, v, n=1)
models/utils.py:152
↓ 21 callersMethodget_height
(self)
bricks/brick_info.py:962
↓ 21 callersMethodsave
save the current content to the HMTL file
util/html.py:69
↓ 20 callersFunctionget_brick_class
(bid)
bricks/brick_info.py:577
↓ 17 callersMethodadd_brick
op_type: 0: new brick's astud -> current bricks' stud 1: new brick's stud -> current bricks' astud 2: put on the ground
bricks/brick_info.py:1243
↓ 15 callersFunction_nms
(heat, kernel=3)
models/heatmap/models/decode.py:11
↓ 14 callersFunctiontransform_points_round
(pts, matrix)
bricks/brick_info.py:785
↓ 13 callersFunctionconv3d
( in_channels: int, out_channels: int, kernel_size: int = 3, stride: int = 1,
models/networks.py:228
↓ 12 callersFunction_topk
(scores, K=40)
models/heatmap/models/decode.py:115
↓ 12 callersFunctionadd_cbrick_to_bricks_pc
(bricks_pc: BricksPC, brick: CBrick, op_type=0, verbose=False, debug=False, only_check=False,
bricks/brick_info.py:1716
↓ 12 callersFunctionget_all_brick_ids
()
bricks/brick_info.py:563
↓ 12 callersMethodstate_dict
(self)
models/utils.py:20
↓ 11 callersFunctionget_rank
r"""Get rank of the thread.
tu/ddp.py:133
↓ 11 callersFunctionid2euler
(idx, symmetry_aware_label=False)
datasets/legokps_shape_cond_dataset.py:71
↓ 10 callersFunctionblend_img
(back, fore, theme='black', trans=0.7)
models/heatmap/utils/vis.py:115
↓ 10 callersFunctiongen_colormap
(img, theme='black', output_res=None, colors=colors)
models/heatmap/utils/vis.py:97
↓ 10 callersFunctionremove_np_arrays
(l: List[np.array], arrs: List[np.array])
bricks/brick_info.py:216
↓ 9 callersFunctiondict_to_cbrick
Reset_pose is used when you want a canonical cbrick.
bricks/brick_info.py:1656
↓ 9 callersMethodload_state_dict
(self, state_dict)
models/utils.py:26
↓ 8 callersMethod__init__
(self)
models/heatmap/models/losses.py:184
↓ 8 callersMethoddump
(cls, layout: List[List[Any]], table_name='')
tu/test_utils.py:33
↓ 8 callersFunctionexpand_mask
(m, num_classes=-1)
models/utils.py:97
↓ 8 callersMethodfrom_dict
(d, no_check=False)
bricks/brick_info.py:1527
↓ 8 callersFunctionget_brick_info
Args: brick: grid_size: Returns: occ_pts, stud_pts, astud_pts, base_contour
bricks/brick_info.py:585
↓ 7 callersFunctionmaster_only_print
r"""master-only print
tu/ddp.py:171
↓ 7 callersMethodrow
(self, *args, **kwargs)
util/html_table.py:112
↓ 6 callersMethod__init__
(self)
models/heatmap/lego_hg.py:427
↓ 6 callersMethodgather
(self, outputs, output_device)
models/heatmap/models/data_parallel.py:83
↓ 6 callersFunctionget_cbrick_keypoint
:param policy: if 'simple', uses origin + [0, height, 0] as the origin. If 'brick', select a brick with the greatest height and use it's
bricks/brick_info.py:1697
↓ 6 callersMethodtable
(self, name, columns)
util/html_table.py:81
↓ 6 callersFunctionupdate_with_assertion
Assert that all_brick_list doesn't have any bricks in brick_list before being udpated.
bricks/brick_info.py:529
↓ 5 callersMethod__init__
( self, n, nstack, dims, modules, heads, pre=None, cnv_dim=256, make_tl_layer=None, ma
models/heatmap/models/networks/hourglass.py:197
↓ 5 callersMethodapply_transform
(self, matrix)
bricks/brick_info.py:1188
↓ 5 callersFunctionchamfer_distance
Chamfer distance between two point clouds Parameters ---------- x: numpy array [n_points_x, n_dims] first point cloud y: numpy
bricks/utils.py:96
↓ 5 callersMethodcreate_visdom_connections
If the program could not connect to Visdom server, this function will start a new server at port < self.port >
util/visualizer.py:101
↓ 5 callersMethodget_stud_positions
(self)
bricks/brick_info.py:1442
↓ 5 callersFunctionpts2voxel
(pts, value)
bricks/brick_info.py:696
↓ 5 callersMethodscatter
(self, inputs, kwargs, device_ids, chunk_sizes)
models/heatmap/models/data_parallel.py:77
↓ 5 callersFunctionsimple_brick_info
(l, h, w)
bricks/brick_info.py:79
↓ 5 callersFunctionvisualize_bricks
(bricks: List[Brick], highlight=False, adjust_camera=True)
debug/utils.py:63
↓ 4 callersFunction_h_aggregate
(heat, aggr_weight=0.1)
models/heatmap/models/decode.py:78
↓ 4 callersFunction_v_aggregate
(heat, aggr_weight=0.1)
models/heatmap/models/decode.py:83
↓ 4 callersMethodadd_brick_
(self, brick: Union[Brick, VBrick, HBrick], op_type, verbose, only_check, brick_info=None,
bricks/brick_info.py:1268
↓ 4 callersFunctionbricks2meshes
(bricks: List[Union[Brick, CBrick]], colors)
data_generation/utils.py:132
↓ 4 callersMethodcanonize_link
(self, filetype, obj, row_identifier=None, col_identifier=None)
tu/loggers/html_table.py:268
↓ 4 callersFunctioncompute_res_loss
(output, target)
models/heatmap/models/losses.py:258
↓ 4 callersFunctioncreate_dataset
Create a dataset given the option. This function wraps the class CustomDatasetDataLoader. This is the main interface between this package
datasets/__init__.py:55
↓ 4 callersFunctionevaluate_set
(opt, model, set_path, meters, is_subm=False, subm_cbricks=None, subm_meters=None)
eval.py:122
↓ 4 callersMethodget
Download a dataset. Parameters: save_path (str) -- A directory to save the data to. dataset (str) -- (opt
util/get_data.py:81
↓ 4 callersMethodget_astud_positions
(self)
bricks/brick_info.py:1445
↓ 4 callersFunctionget_brick_enc_voxel_
(bid, position=None, rotation=None, extra_point=None, extra_point_value=None, use_id=False)
bricks/brick_info.py:682
↓ 4 callersFunctionget_cameras
(azim, elev, device='cuda')
lego/utils/camera_utils.py:13
↓ 4 callersFunctionget_coords
(l)
bricks/brick_info.py:888
↓ 4 callersMethodget_current_losses
Return traning losses / errors. train.py will print out these errors on console, and save them to a file
models/base_model.py:146
↓ 4 callersFunctionget_save_path
(name, ext, add_timestamp=False, verbose=True)
debug/utils.py:123
↓ 4 callersFunctionget_world_size
r"""Get world size. How many GPUs are available in this job.
tu/ddp.py:142
↓ 4 callersFunctionline_area
(x0, z0, y, x1, z1)
bricks/utils.py:30
↓ 4 callersMethodmerge_from
(self, m, ks=None)
models/utils.py:165
↓ 4 callersFunctionmkdirs
create empty directories if they don't exist Parameters: paths (str list) -- a list of directory paths
util/util.py:130
↓ 4 callersMethodparse
Parse our options, create checkpoints directory suffix, and set up gpu device.
options/base_options.py:154
↓ 4 callersMethodprint_url
(vis: Union[HTMLTableVisualizer, str], verbose=True)
tu/loggers/html_helper.py:44
↓ 4 callersMethodsave_networks
Save all the networks to the disk. Parameters: epoch (int) -- current epoch; used in the file name '%s_net_%s.pth' % (epoch, name
models/base_model.py:155
↓ 4 callersMethodset_input
Unpack input data from the dataloader and perform necessary pre-processing steps. Parameters: input (dict): includes the data its
models/base_model.py:63
↓ 4 callersFunctionsimple_voxel_encoder
(num_features, num_input_features=1, addconv=True, n_layers=4, avg_pool=True, num_embedding=0,
models/heatmap/lego_hg.py:474
↓ 4 callersMethodstep
(self, closure=None)
models/utils.py:33
↓ 4 callersMethodto_dict
(self)
bricks/brick_info.py:997
↓ 3 callersMethod__init__
brick_type: specify the shape of the brick (like 1x1x2, 1x1xl..) position size: # rotation: [0-3] counterclockwise, 0
bricks/brick_info.py:909
↓ 3 callersMethod__init__
(self, pretrain=False, use_layer2=False)
models/networks.py:142
↓ 3 callersMethod_get_brick_transform
(self, brick: Brick, canonical)
bricks/brick_info.py:1458
↓ 3 callersMethodadd_header
Insert a header to the HTML file Parameters: text (str) -- the header text
util/html.py:40
↓ 3 callersMethodadd_images
add images to the HTML file Parameters: ims (str list) -- a list of image paths txts (str list) -- a list of image
util/html.py:49
↓ 3 callersFunctionapply_mask
Apply the given mask to the image.
coco_related/visualize.py:72
↓ 3 callersMethodavg_dict
(self)
models/utils.py:159
↓ 3 callersFunctionbrick2p3dmesh
(brick, color=None, cuda=False)
data_generation/utils.py:107
↓ 3 callersFunctioncbrick_highest_brick
(cbrick: CBrick, return_ind=False, allow_multiple=False)
bricks/brick_info.py:1667
↓ 3 callersMethodcompute_instance_masks
(self, kmeans_init_kps=None, reorder_idxs_list=None)
models/hourglass_shape_cond_model.py:238
↓ 3 callersMethodcompute_visuals
Calculate additional output images for visdom and HTML visualization
models/base_model.py:118
↓ 3 callersFunctioncreate_model
Create a model given the option. This function warps the class CustomDatasetDataLoader. This is the main interface between this package and '
models/__init__.py:56
↓ 3 callersFunctioncreate_window
(window_size, channel=1)
tu/metrics/msssim.py:12
↓ 3 callersMethodeval
Make models eval mode during test time
models/base_model.py:101
↓ 3 callersFunctionfn
(mask_color)
data_generation/utils.py:264
↓ 3 callersMethodget_asset_filename
(self, row_identifier, col_identifier, ext)
tu/loggers/html_table.py:242
↓ 3 callersFunctionget_brick_annotation
(bid)
bricks/brick_info.py:21
↓ 3 callersFunctionget_brick_enc_voxel_info
()
bricks/brick_info.py:649
↓ 3 callersMethodget_brick_info
(self, grid_size)
bricks/brick_info.py:970
↓ 3 callersFunctionget_brick_info_raw
(bid)
bricks/brick_info.py:581
↓ 3 callersFunctionget_brick_masks
(bricks_mesh_list, mask_colors, idxs, cameras, render_size=1536, output_size=512)
data_generation/utils.py:209
↓ 3 callersFunctionget_brick_name
(bid)
test_kp_shape_cond.py:274
↓ 3 callersFunctionget_connection_offset
(brick_type_or_cbrick, rotation_quat, op_type)
bricks/brick_info.py:800
↓ 3 callersMethodget_current_visuals
Return visualization images. train.py will display these images with visdom, and save the images to a HTML
models/base_model.py:138
↓ 3 callersFunctionindex_list
(l, idxs)
debug/utils.py:319
↓ 3 callersFunctionis_dist_avail_and_initialized
()
coco_related/utils.py:271
↓ 3 callersFunctionoffset_pts
(pts: Union[List, Set], *, offset_x=0, offset_h=0, offset_y=0)
bricks/utils.py:15
next →1–100 of 675, ranked by callers