MCPcopy Create free account

hub / github.com/EryiXie/PlaneRecNet / functions

Functions218 in github.com/EryiXie/PlaneRecNet

↓ 60 callersMethodappend
Same as add just more pythonic.
utils/utils.py:31
↓ 24 callersMethodcopy
Copies this config into a new config object, making the changes given by new_config_dict.
data/config.py:55
↓ 19 callersMethodreplace
Copies new_config_dict into this config object. Note: new_config_dict can also be a config object.
data/config.py:68
↓ 8 callersMethodget_avg
Returns the average of the elements in the window.
utils/utils.py:40
↓ 8 callersFunctionset_cfg
Sets the active config. Works even if cfg is already imported!
data/config.py:533
↓ 6 callersMethodadd
Adds an element to the window, removing the earliest element if necessary.
utils/utils.py:19
↓ 6 callersFunctionenable_if
(condition, obj)
data/augmentations.py:460
↓ 5 callersMethod_make_layer
Here one layer means a string of n Bottleneck blocks.
models/backbone.py:144
↓ 4 callersMethod__init__
(self)
models/functions/losses.py:293
↓ 4 callersMethodload_weights
Loads weights from a compressed save file.
planerecnet.py:125
↓ 4 callersFunctionreplace
(name)
train.py:100
↓ 4 callersMethodreset
Resets the MovingAverage to its initial state.
utils/utils.py:35
↓ 3 callersMethod__init__
(self, cfg)
planerecnet.py:21
↓ 3 callersMethodfreeze_bn
Adapted from https://discuss.pytorch.org/t/how-to-train-with-frozen-batchnorm/12106/8
planerecnet.py:147
↓ 3 callersMethodfrom_str
(path:str)
utils/utils.py:119
↓ 3 callersMethodget_latest
Note: config should be config.name.
utils/utils.py:148
↓ 3 callersFunctioninference_image
(net: PlaneRecNet, path: str, save_path: str = None, depth_mode: str='colored')
simple_inference.py:143
↓ 3 callersMethodinit_weights
Initialize weights for training.
planerecnet.py:130
↓ 3 callersMethodnormal_from_triplets
(self, triplets, simpled_mask)
models/functions/vnl.py:106
↓ 3 callersMethodpull_item
instances: Dict {'masks': [N x H x W], torch.uint8, 'boxes': [N x 4], torch.float64, 'classe
data/datasets.py:52
↓ 3 callersMethodsave_weights
Saves the model's weights using compression because the file sizes were getting too big.
planerecnet.py:121
↓ 3 callersFunctionset_dataset
Sets the dataset of the current config.
data/config.py:543
↓ 2 callersMethod__init__
(self, image_path, anno_file, transform=None, dataset_name=None, has_gt=True, has_pos=True)
data/datasets.py:212
↓ 2 callersMethod_update_str
(self)
utils/utils.py:78
↓ 2 callersFunctioncompute_validation_metrics
(epoch, iteration, prn_net, val_dataset, eval_nums=-1)
train.py:440
↓ 2 callersFunctiondisplay_on_frame
(result, frame, mask_alpha=0.5, fps_str='', no_mask=False, no_box=False, no_text=False)
simple_inference.py:49
↓ 2 callersMethodfilter_mask
(self, p123, point_cloud, delta_cos=0.985, delta_diff=0.005)
models/functions/vnl.py:74
↓ 2 callersMethodform_pw_groups
Form 3D points groups, with 3 points in each grouup. :param p123: points index :param pw: 3D points :return:
models/functions/vnl.py:57
↓ 2 callersMethodform_pw_groups
Form 3D points groups, with 3 points in each grouup. :param p123: points index :param pw: 3D points :return:
models/functions/vnl.py:235
↓ 2 callersFunctionget_color
(j, on_gpu=None)
simple_inference.py:64
↓ 2 callersMethodget_depth_path
(self, rgb_file_name)
data/datasets.py:171
↓ 2 callersMethodget_interrupt
(save_folder)
utils/utils.py:142
↓ 2 callersFunctionimresize
Resize image to a given size. Args: img (ndarray): The input image. size (tuple[int]): Target size (w, h). return_scale (b
models/functions/funcs.py:95
↓ 2 callersFunctionmake_row
(vals)
eval.py:358
↓ 2 callersFunctionmake_sep
(n)
eval.py:359
↓ 2 callersMethodpush
(self, score: float, is_true: bool)
eval.py:264
↓ 2 callersMethodselect_index
(self, num)
models/functions/vnl.py:43
↓ 2 callersFunctionset_lr
(optimizer, new_lr)
train.py:415
↓ 2 callersFunctionstart
Start timing the specific function. Note: If use_stack is True, only one timer can be active at a time. Once you stop this timer, the previo
utils/timer.py:54
↓ 2 callersFunctionstop
If use_stack is True, this will stop the currently running timer and restore the previous timer on the stack if that exists. Note if use_stack is T
utils/timer.py:76
↓ 2 callersMethodtransfer_xyz
(self, depth, k_maritix)
models/functions/vnl.py:34
↓ 2 callersMethodtransfer_xyz
(self, depth, fx, fy)
models/functions/vnl.py:206
↓ 1 callersFunctionPCA_svd
(pts)
models/functions/funcs.py:287
↓ 1 callersMethod__init__
(self, width, height, mean=MEANS, pad_gt=True)
data/augmentations.py:112
↓ 1 callersMethod__init__
( self, inplanes, planes, stride=1, downsample=None, norm_laye
models/backbone.py:10
↓ 1 callersMethod__init__
(self, input_size, delta_cos=0.867, delta_z=0.0001, sample_ratio=0.3)
models/functions/vnl.py:7
↓ 1 callersMethod__len__
(self)
data/datasets.py:119
↓ 1 callersFunction_scale_size
Rescale a size by a ratio. Args: size (tuple[int]): (w, h). scale (float): Scaling factor. Returns: tuple[int]: scaled
models/functions/funcs.py:74
↓ 1 callersMethodadd_gt_positives
Call this once per image.
eval.py:267
↓ 1 callersMethodadd_layer
Add a downsample layer to the backbone as per what SSD does.
models/backbone.py:226
↓ 1 callersFunctionbbox_iou
Compute the jaccard overlap of two sets of boxes. The jaccard overlap is simply the intersection over union of two boxes. Here we operate on
models/functions/funcs.py:30
↓ 1 callersFunctionbias_init_with_prob
initialize conv/fc bias value according to giving probablity
models/functions/funcs.py:329
↓ 1 callersFunctioncalc_map
(ap_data)
eval.py:327
↓ 1 callersFunctioncalc_size_preserve_ar
(img_w, img_h, max_size)
models/functions/funcs.py:195
↓ 1 callersFunctioncenter_of_mass
(bitmasks)
models/functions/funcs.py:213
↓ 1 callersFunctioncompute_depth_metrics
Computation of error metrics between predicted and ground truth depths. Prediction and ground turth need to be converted to the same unit e.g
eval.py:164
↓ 1 callersFunctioncompute_gradient_map
Compute gradient map from depth map with 3x3 sobel filter
models/functions/losses.py:305
↓ 1 callersFunctioncompute_segmentation_metrics
(ap_data, gt_masks, gt_boxes, gt_classes, pred_masks, pred_boxes, pred_classes, pred_scores)
eval.py:210
↓ 1 callersFunctionconstruct_backbone
Constructs a backbone given a backbone config object (see config.py).
models/backbone.py:233
↓ 1 callersFunctionenforce_size
Ensures that the image is the given size without distorting aspect ratio.
data/datasets.py:276
↓ 1 callersFunctionevaluate
(net: PlaneRecNet, dataset, during_training=False, eval_nums=-1)
eval.py:63
↓ 1 callersMethodfilter_mask
(self, p123, gt_xyz, delta_cos=0.867, delta_diff_x=0.005, delta_diff_y
models/functions/vnl.py:256
↓ 1 callersMethodget_ap
Warning: result not cached.
eval.py:274
↓ 1 callersMethodget_camera_matrix
(self, rgb_file_name)
data/datasets.py:174
↓ 1 callersFunctionget_label_map
()
data/datasets.py:15
↓ 1 callersMethodget_path
(self, root:str='')
utils/utils.py:114
↓ 1 callersMethodget_plane_para
(self, target)
data/datasets.py:177
↓ 1 callersFunctionget_points_coordinate
(depth, intrinsic_inv)
models/functions/funcs.py:226
↓ 1 callersFunctionibims1
(net: PlaneRecNet, in_folder: str, out_folder: str)
simple_inference.py:202
↓ 1 callersFunctionibims1_pd
(net: PlaneRecNet, in_folder: str, out_folder: str)
simple_inference.py:240
↓ 1 callersFunctionimrescale
Resize image while keeping the aspect ratio. Args: img (ndarray): The input image. scale (float | tuple[int]): The scaling factor
models/functions/funcs.py:173
↓ 1 callersMethodinference
(self, pred_masks, pred_cates, pred_kernels, pred_depths, batched_images)
planerecnet.py:155
↓ 1 callersFunctioninference_images
(net: PlaneRecNet, in_folder: str, out_folder: str, max_img: int=0, depth_mode: str='colored')
simple_inference.py:180
↓ 1 callersMethodinference_single_image
(self, seg_preds, cate_preds, kernel_preds, depth_pred, ori_size)
planerecnet.py:182
↓ 1 callersMethodinit_backbone
Initializes the backbone weights for training.
models/backbone.py:211
↓ 1 callersFunctioninit_console
Initialize the console to be able to use ANSI escape characters on Windows.
utils/utils.py:92
↓ 1 callersMethodinit_image_coor
(self)
models/functions/vnl.py:19
↓ 1 callersMethodinit_image_coor
(self)
models/functions/vnl.py:191
↓ 1 callersFunctionintersect
We resize both tensors to [A,B,2] without new malloc: [A,2] -> [A,1,2] -> [A,B,2] [B,2] -> [1,B,2] -> [A,B,2] Then we compute the area of
models/functions/funcs.py:9
↓ 1 callersMethodis_empty
(self)
eval.py:271
↓ 1 callersFunctionlog_losses
Write losses to the event file
train.py:451
↓ 1 callersFunctionlog_visual_example
Write visaul examples to the event file
train.py:463
↓ 1 callersFunctionmask_iou
Computes the pariwise mask IoU between two sets of masks of size [a, h, w] and [b, h, w]. The output is of size [a, b].
models/functions/funcs.py:58
↓ 1 callersFunctionmask_nms
(cate_labels, seg_masks, sum_masks, cate_scores, nms_thr=0.5)
models/functions/nms.py:53
↓ 1 callersFunctionmatrix_nms
(cate_labels, seg_masks, sum_masks, cate_scores, sigma=2.0, kernel='gaussian')
models/functions/nms.py:15
↓ 1 callersFunctionpad_even_divided
(img, divisor=32)
models/functions/funcs.py:204
↓ 1 callersFunctionparse_args
(argv=None)
eval.py:30
↓ 1 callersFunctionparse_args
(argv=None)
planerecnet.py:612
↓ 1 callersFunctionparse_args
(argv=None)
simple_inference.py:25
↓ 1 callersFunctionparse_args
(argv=None)
data/datasets.py:326
↓ 1 callersFunctionpoint_nms
(heat, kernel=2)
models/functions/nms.py:8
↓ 1 callersMethodprepare_data
(self, datum, devices:list=None, allocation:list=None)
train.py:177
↓ 1 callersMethodprepare_ground_truth
(self, gt_instances_per_frame, mask_feat_size)
models/functions/losses.py:201
↓ 1 callersMethodprint
(self)
data/config.py:79
↓ 1 callersFunctionprint_maps
(all_maps)
eval.py:356
↓ 1 callersMethodpull_image
Returns the original image object at index in OPENCV form (BGR) Argument: index (int): index of img to show Return:
data/datasets.py:123
↓ 1 callersMethodremove_interrupt
(save_folder)
utils/utils.py:137
↓ 1 callersFunctionrescale_size
Calculate the new size to be rescaled to. Args: old_size (tuple[int]): The old size (w, h) of image. scale (float | tuple[int]): T
models/functions/funcs.py:138
↓ 1 callersMethodselect_index
(self)
models/functions/vnl.py:213
↓ 1 callersMethodselect_points_groups
(self, gt_depth, pred_depth, fx, fy)
models/functions/vnl.py:291
next →1–100 of 218, ranked by callers