MCPcopy Create free account

hub / github.com/Relento/lego_release / functions

Functions675 in github.com/Relento/lego_release

Functionnamed_timeit
(name, store_dict)
tu/loggers/timer.py:13
Functionnamed_timeit
(name, store_dict)
data_generation/utils.py:313
Functionndc_meshgrid
(size)
models/networks.py:298
Functionnested_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
Methodnewline
(self)
tu/configs.py:459
Functionoffset_boxes
(bl, h, offset_x, offset_y)
bricks/utils.py:45
Methodoptimize_parameters
Update network weights; it will be called in every training iteration.
models/hourglass_shape_cond_model.py:729
Methodoptimize_parameters
Update network weights; it will be called in every training iteration.
models/hourglass_trans_model.py:496
Methodparam_groups
(self)
models/utils.py:17
Methodplot_current_losses
display the current losses on visdom display: dictionary of error labels and values Parameters: epoch (int) -- current
tu/loggers/visualizer.py:49
Functionplot_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
Functionplot_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
Methodprint_button
(vis: HTMLTableVisualizer, prev_vis=None, prev_button=None, next_vis=None, next_button=None)
tu/loggers/html_helper.py:54
Methodprint_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
Functionprint_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
Methodprint_over
Remove anything printed in the last printover call. Then print `string`
tu/configs.py:449
Functionprint_time
(name)
tu/loggers/timer.py:6
Methodquery
Return an image from the pool. Parameters: images: the latest generated images from the generator Returns images from th
util/image_pool.py:24
Functionrecursive_items
Get all (key, item) recursively in a potentially recursive dictionary. Usage: ``` x = { 'foo' : { 'bar' : 5
tu/configs.py:225
Functionreduce_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
Functionreduce_nested_list
(reduce_fn, nested_list, template)
lego/utils/data_generation_utils.py:48
Functionremap_idxs
(i)
models/utils.py:122
Functionremove_batchnorm
(module)
models/hourglass_shape_cond_model.py:30
Functionremove_batchnorm
(module)
models/hourglass_trans_model.py:28
Methodreset
Reset the self.saved status
tu/loggers/visualizer.py:28
Functionresize_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
Methodreverse_id
(self, int_id)
tu/configs.py:391
Functionround_rectangle
Draw a rounded rectangle
data_generation/utils.py:28
Functionsaturation_
(data_rng, image, gs, gs_mean, var)
models/heatmap/utils/image.py:212
Functionsave_brick
(brick, add_timestamp=False, as_dict=True, name='brick')
debug/utils.py:232
Functionsave_bricks_image
(bricks: List[Brick], add_timestamp=False, highlight=False, name='bricks')
debug/utils.py:261
Functionsave_bricks_pc
(bs: BricksPC, add_timestamp=False, as_dict=True, name='bs', info=None)
debug/utils.py:179
Functionsave_bricks_pc_image
(bs: BricksPC, add_timestamp=False, highlight=False, name='bs')
debug/utils.py:218
Functionsave_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
Functionsave_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
Functionsave_model
(path, epoch, model, optimizer=None)
models/heatmap/models/model.py:86
Methodsave_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
Methodsave_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
Functionsave_on_master
(*args, **kwargs)
coco_related/utils.py:295
MethodsetUpClass
(cls)
tu/test_utils.py:18
Functionset_consider_symmetry
(v)
bricks/brick_info.py:51
Methodset_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
Methodset_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
Methodset_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
Methodsetup
(self, opt)
models/hourglass_shape_cond_model.py:180
Methodsetup
(self, opt)
models/hourglass_trans_model.py:182
Functionsimple_voxel_decoder
(num_features, num_output_features=1, n_layers=4)
models/heatmap/lego_hg.py:496
Functionsimple_voxel_encoder1
(num_features, num_input_features=1, addconv=True)
models/networks.py:274
Functionsimple_voxel_encoder2
(num_features, num_input_features=3, addconv=True)
models/networks.py:286
Functionspawn_ddp
Args: worker: a function with argument rank, world_size, args_in example see test_ddp_spawn Returns:
tu/train_setup.py:18
Methodstate
(self)
models/utils.py:13
Methodstud_leave_vacant
(self)
bricks/brick_info.py:1616
Methodsummarize
(self)
coco_related/coco_eval.py:53
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
coco_related/utils.py:30
Methodsynchronize_between_processes
(self)
coco_related/utils.py:172
Methodsynchronize_between_processes
(self)
coco_related/coco_eval.py:44
Functiontakespread
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
MethodtearDownClass
(cls)
tu/test_utils.py:29
Functiontensor2flow
r"""Convert flow tensor to color image. Args: tensor (tensor) of If tensor then (NxCxHxW) or (NxTxCxHxW) or (CxHxW). imty
tu/flow.py:7
Functiontensor2im
Args: image_tensor: (list of) shape [..., C, H, W] Returns:
tu/loggers/utils.py:8
Functiontensor2im
Converts a Tensor array into a numpy image array. Parameters: input_image (tensor) -- the input image tensor array imtype (type)
util/util.py:63
FunctiontoBbox
(rleObjs)
data_generation/utils.py:76
Methodto_dict
(self)
bricks/brick_info.py:1105
Methodto_dict
(self, bricks_nums=None)
bricks/brick_info.py:1495
Methodto_dict
(self)
bricks/brick_info.py:1588
Functiontransform_preds
(coords, center, scale, output_size)
models/heatmap/utils/image.py:15
Methodupdate
(self, **kwargs)
coco_related/utils.py:149
Methodupdate
(self, predictions)
coco_related/coco_eval.py:29
Methodvalue
(self)
coco_related/utils.py:62
Functionvisualize_brick
(b_or_b_type: Union[str, Brick], highlight=False, adjust_camera=True)
debug/utils.py:113
Functionvisualize_brick_info
(t, b, s, a, c)
bricks/brick_info.py:883
Functionvisulize_flow_file
(flow_filename, save_dir=None)
tu/flow.py:104
Functionwarmup_lr_scheduler
(optimizer, warmup_iters, warmup_factor)
coco_related/utils.py:238
Functionwrapper
r"""Simple function wrapper for the master function
tu/ddp.py:155
FunctionwriteFlow
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
← previous601–675 of 675, ranked by callers