Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Relento/lego_release
/ functions
Functions
675 in github.com/Relento/lego_release
⨍
Functions
675
◇
Types & classes
88
Function
named_timeit
(name, store_dict)
tu/loggers/timer.py:13
Function
named_timeit
(name, store_dict)
data_generation/utils.py:313
Function
ndc_meshgrid
(size)
models/networks.py:298
Function
nested_dict_to_dot_map_dict
Convert a recursive dictionary into a flat, dot-map dictionary. :param d: e.g. {'a': {'b': 2, 'c': 3}} :param parent_key: Used for recur
tu/configs.py:95
Method
newline
(self)
tu/configs.py:459
Function
offset_boxes
(bl, h, offset_x, offset_y)
bricks/utils.py:45
Method
optimize_parameters
Update network weights; it will be called in every training iteration.
models/hourglass_shape_cond_model.py:729
Method
optimize_parameters
Update network weights; it will be called in every training iteration.
models/hourglass_trans_model.py:496
Method
param_groups
(self)
models/utils.py:17
Method
plot_current_losses
display the current losses on visdom display: dictionary of error labels and values Parameters: epoch (int) -- current
tu/loggers/visualizer.py:49
Function
plot_overlaps
Draw a grid showing how ground truth objects are classified. gt_class_ids: [N] int. Ground truth class IDs pred_class_id: [N] int. Predicted c
coco_related/visualize.py:320
Function
plot_precision_recall
Draw the precision-recall curve. AP: Average precision at IoU >= 0.5 precisions: list of precision values recalls: list of recall values
coco_related/visualize.py:305
Method
print_button
(vis: HTMLTableVisualizer, prev_vis=None, prev_button=None, next_vis=None, next_button=None)
tu/loggers/html_helper.py:54
Method
print_current_losses
print current losses on console; also save the losses to the disk Parameters: epoch (int) -- current epoch iters (int
tu/loggers/visualizer.py:75
Function
print_numpy
Print the mean, min, max, median, std, and size of a numpy array Parameters: val (bool) -- if print the values of the numpy array
util/util.py:114
Method
print_over
Remove anything printed in the last printover call. Then print `string`
tu/configs.py:449
Function
print_time
(name)
tu/loggers/timer.py:6
Method
query
Return an image from the pool. Parameters: images: the latest generated images from the generator Returns images from th
util/image_pool.py:24
Function
recursive_items
Get all (key, item) recursively in a potentially recursive dictionary. Usage: ``` x = { 'foo' : { 'bar' : 5
tu/configs.py:225
Function
reduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
coco_related/utils.py:117
Function
reduce_nested_list
(reduce_fn, nested_list, template)
lego/utils/data_generation_utils.py:48
Function
remap_idxs
(i)
models/utils.py:122
Function
remove_batchnorm
(module)
models/hourglass_shape_cond_model.py:30
Function
remove_batchnorm
(module)
models/hourglass_trans_model.py:28
Method
reset
Reset the self.saved status
tu/loggers/visualizer.py:28
Function
resize_masks
Resize masks size :param masks: numpy of shape (n, 1, h, w) :param image_size: H, W :return: numpy array of shape (n, H, W)
util/util.py:15
Method
reverse_id
(self, int_id)
tu/configs.py:391
Function
round_rectangle
Draw a rounded rectangle
data_generation/utils.py:28
Function
saturation_
(data_rng, image, gs, gs_mean, var)
models/heatmap/utils/image.py:212
Function
save_brick
(brick, add_timestamp=False, as_dict=True, name='brick')
debug/utils.py:232
Function
save_bricks_image
(bricks: List[Brick], add_timestamp=False, highlight=False, name='bricks')
debug/utils.py:261
Function
save_bricks_pc
(bs: BricksPC, add_timestamp=False, as_dict=True, name='bs', info=None)
debug/utils.py:179
Function
save_bricks_pc_image
(bs: BricksPC, add_timestamp=False, highlight=False, name='bs')
debug/utils.py:218
Function
save_image
Save a numpy image to the disk Parameters: image_numpy (numpy array) -- input numpy array image_path (str) -- the path o
util/util.py:103
Function
save_images
Save images to the disk. Parameters: webpage (the HTML class) -- the HTML webpage class that stores these imaegs (see html.py for more de
util/visualizer.py:19
Function
save_model
(path, epoch, model, optimizer=None)
models/heatmap/models/model.py:86
Method
save_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/hourglass_shape_cond_model.py:749
Method
save_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/hourglass_trans_model.py:516
Function
save_on_master
(*args, **kwargs)
coco_related/utils.py:295
Method
setUpClass
(cls)
tu/test_utils.py:18
Function
set_consider_symmetry
(v)
bricks/brick_info.py:51
Method
set_input
Unpack input data from the dataloader and perform necessary pre-processing steps. Parameters: input: a dictionary that contains t
models/hourglass_shape_cond_model.py:183
Method
set_input
Unpack input data from the dataloader and perform necessary pre-processing steps. Parameters: input: a dictionary that contains t
models/hourglass_trans_model.py:185
Method
set_requires_grad
Set requies_grad=Fasle for all the networks to avoid unnecessary computations Parameters: nets (network list) -- a list of netwo
models/base_model.py:216
Method
setup
(self, opt)
models/hourglass_shape_cond_model.py:180
Method
setup
(self, opt)
models/hourglass_trans_model.py:182
Function
simple_voxel_decoder
(num_features, num_output_features=1, n_layers=4)
models/heatmap/lego_hg.py:496
Function
simple_voxel_encoder1
(num_features, num_input_features=1, addconv=True)
models/networks.py:274
Function
simple_voxel_encoder2
(num_features, num_input_features=3, addconv=True)
models/networks.py:286
Function
spawn_ddp
Args: worker: a function with argument rank, world_size, args_in example see test_ddp_spawn Returns:
tu/train_setup.py:18
Method
state
(self)
models/utils.py:13
Method
stud_leave_vacant
(self)
bricks/brick_info.py:1616
Method
summarize
(self)
coco_related/coco_eval.py:53
Method
synchronize_between_processes
Warning: does not synchronize the deque!
coco_related/utils.py:30
Method
synchronize_between_processes
(self)
coco_related/utils.py:172
Method
synchronize_between_processes
(self)
coco_related/coco_eval.py:44
Function
takespread
Get `num` elements from the sequence that are as spread out as possible. https://stackoverflow.com/questions/9873626/choose-m-evenly-spaced-
tu/configs.py:325
Method
tearDownClass
(cls)
tu/test_utils.py:29
Function
tensor2flow
r"""Convert flow tensor to color image. Args: tensor (tensor) of If tensor then (NxCxHxW) or (NxTxCxHxW) or (CxHxW). imty
tu/flow.py:7
Function
tensor2im
Args: image_tensor: (list of) shape [..., C, H, W] Returns:
tu/loggers/utils.py:8
Function
tensor2im
Converts a Tensor array into a numpy image array. Parameters: input_image (tensor) -- the input image tensor array imtype (type)
util/util.py:63
Function
toBbox
(rleObjs)
data_generation/utils.py:76
Method
to_dict
(self)
bricks/brick_info.py:1105
Method
to_dict
(self, bricks_nums=None)
bricks/brick_info.py:1495
Method
to_dict
(self)
bricks/brick_info.py:1588
Function
transform_preds
(coords, center, scale, output_size)
models/heatmap/utils/image.py:15
Method
update
(self, **kwargs)
coco_related/utils.py:149
Method
update
(self, predictions)
coco_related/coco_eval.py:29
Method
value
(self)
coco_related/utils.py:62
Function
visualize_brick
(b_or_b_type: Union[str, Brick], highlight=False, adjust_camera=True)
debug/utils.py:113
Function
visualize_brick_info
(t, b, s, a, c)
bricks/brick_info.py:883
Function
visulize_flow_file
(flow_filename, save_dir=None)
tu/flow.py:104
Function
warmup_lr_scheduler
(optimizer, warmup_iters, warmup_factor)
coco_related/utils.py:238
Function
wrapper
r"""Simple function wrapper for the master function
tu/ddp.py:155
Function
writeFlow
Write optical flow to file. If v is None, uv is assumed to contain both u and v channels, stacked in depth. Original code by Deqing Sun,
tu/flow.py:70
← previous
601–675 of 675, ranked by callers