MCPcopy Create free account

hub / github.com/WisconsinAIVision/yolact_edge / functions

Functions396 in github.com/WisconsinAIVision/yolact_edge

↓ 126 callersMethodappend
Same as add just more pythonic.
yolact_edge/utils/functions.py:27
↓ 51 callersMethodcopy
Copies this config into a new config object, making the changes given by new_config_dict.
yolact_edge/data/config.py:103
↓ 21 callersMethodget_avg
Returns the average of the elements in the window.
yolact_edge/utils/functions.py:36
↓ 14 callersMethod__init__
(self, num_layers)
yolact_edge/yolact.py:819
↓ 12 callersMethodadd
Adds an element to the window, removing the earliest element if necessary.
yolact_edge/utils/functions.py:15
↓ 12 callersFunctionenable_if
(condition, obj)
yolact_edge/utils/augmentations.py:893
↓ 10 callersMethodreplace
Copies new_config_dict into this config object. Note: new_config_dict can also be a config object.
yolact_edge/data/config.py:116
↓ 10 callersMethodt
(fea, layer_idx)
yolact_edge/yolact.py:1370
↓ 8 callersMethod__init__
(self, in_planes, out_planes, kernel_size=3, stride=1, groups=1, activation=nn.ReLU6(inplace=True))
yolact_edge/backbone.py:445
↓ 8 callersMethoddump
(self)
eval.py:322
↓ 8 callersMethodhas_trt_cached_module
(self, module_name, int8_mode=False, batch_size=1)
yolact_edge/yolact.py:1459
↓ 8 callersMethodset_val
(self, new_val)
yolact_edge/utils/functions.py:58
↓ 8 callersMethodtrt_load_if
(self, module_name, trt_fn, trt_fn_params, int8_mode=False, parent=None, batch_size=1)
yolact_edge/yolact.py:1475
↓ 7 callersFunctionrender
()
yolact_edge/scripts/make_grid.py:27
↓ 7 callersMethodreset
Resets the MovingAverage to its initial state.
yolact_edge/utils/functions.py:31
↓ 6 callersMethod_make_layer
Here one layer means a string of n blocks.
yolact_edge/backbone.py:272
↓ 6 callersFunctionbuild_flow_convs
(encode_layers, in_features, out_features, stride=1, groups=1)
yolact_edge/yolact.py:665
↓ 6 callersFunctionsanitize_coordinates
Sanitizes the input coordinates so that x1 < x2, x1 != x2, x1 >= 0, and x2 <= image_size. Also converts from relative to absolute coordinates
yolact_edge/layers/box_utils.py:283
↓ 5 callersMethod_make_layer
Here one layer means a string of n Bottleneck blocks.
yolact_edge/backbone.py:84
↓ 5 callersMethodadd_scalar
(self, key, value)
yolact_edge/utils/tensorboard_helper.py:11
↓ 5 callersFunctionconv_lrelu
(in_features, out_features, kernel_size=3, stride=1, dilation=1, groups=1)
yolact_edge/yolact.py:448
↓ 5 callersFunctionjaccard
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
yolact_edge/layers/box_utils.py:54
↓ 5 callersFunctionpostprocess
Postprocesses the output of Yolact on testing mode into a format that makes sense, accounting for all the possible configuration settings.
yolact_edge/layers/output_utils.py:15
↓ 5 callersFunctionprep_display
Note: If undo_transform=False then im_h and im_w are allowed to be None.
eval.py:155
↓ 5 callersFunctionset_cfg
Sets the active config. Works even if cfg is already imported!
yolact_edge/data/config.py:946
↓ 4 callersFunctionconvert_image
(image)
train.py:538
↓ 4 callersFunctiondarknetconvlayer
Implements a conv, activation, then batch norm. Arguments are passed into the conv layer.
yolact_edge/backbone.py:211
↓ 4 callersFunctionget_color
(j, on_gpu=None)
eval.py:190
↓ 4 callersFunctiongrabMAP
(string)
yolact_edge/scripts/parse_eval.py:20
↓ 4 callersFunctionprepare_data
(datum)
train.py:644
↓ 4 callersFunctionpretty_str
(x:list)
yolact_edge/scripts/optimize_bboxes.py:157
↓ 4 callersFunctionreplace
(name)
train.py:106
↓ 4 callersFunctionsmoother
(y, interval=100)
yolact_edge/scripts/plot_loss.py:35
↓ 3 callersMethod_get_trt_cache_path
(self, module_name, int8_mode=False, batch_size=1)
yolact_edge/yolact.py:1456
↓ 3 callersFunction_make_divisible
Adapted from torchvision.models.mobilenet._make_divisible.
yolact_edge/backbone.py:456
↓ 3 callersFunctionbackward_and_log
(prefix, net_outs, targets, masks, num_crowds, extra_loss=None)
train.py:321
↓ 3 callersFunctioncalc_map
(ap_data)
eval.py:1164
↓ 3 callersFunctioncenter_size
Convert prior_boxes to (cx, cy, w, h) representation for comparison to center-size form ground truth data. Args: boxes: (tensor) poin
yolact_edge/layers/box_utils.py:20
↓ 3 callersFunctiondecode
Decode predicted bbox coordinates using the same scheme employed by Yolov2: https://arxiv.org/pdf/1612.08242.pdf b_x = (sigmoid(pred
yolact_edge/layers/box_utils.py:223
↓ 3 callersFunctiondeform_op
Warp an image or feature map with optical flow Args: x (Tensor): size (n, c, h, w) flow (Tensor): size (n, 2, h, w), values range
yolact_edge/layers/warp_utils.py:45
↓ 3 callersFunctionevalimage
(net:Yolact, path:str, save_path:str=None, detections:Detections=None, image_id=None)
eval.py:587
↓ 3 callersMethodfaster_rcnn_scale
(width, height, min_size, max_size)
yolact_edge/utils/augmentations.py:185
↓ 3 callersMethodfrom_str
(path:str)
yolact_edge/utils/functions.py:111
↓ 3 callersMethodget_latest
Note: config should be config.name.
yolact_edge/utils/functions.py:140
↓ 3 callersFunctionis_distributed_initialized
()
yolact_edge/utils/misc.py:5
↓ 3 callersMethodload_weights
Loads weights from a compressed save file.
yolact_edge/yolact.py:1183
↓ 3 callersFunctionmake_net
A helper function to take a config setting and turn it into a network. Used by protonet and extrahead. Returns (network, out_channels)
yolact_edge/yolact.py:57
↓ 3 callersFunctionpull_calib_dataset
(calib_folder, transform=transform, max_calibration_images=cfg.torch2trt_max_calibration_images)
yolact_edge/utils/tensorrt.py:54
↓ 3 callersMethodsave_weights
Saves the model's weights using compression because the file sizes were getting too big.
yolact_edge/yolact.py:1179
↓ 3 callersFunctionset_dataset
Sets the dataset of the current config.
yolact_edge/data/config.py:954
↓ 3 callersFunctionset_lr
(optimizer, new_lr)
train.py:622
↓ 3 callersMethodtrain
(self, mode=True)
yolact_edge/yolact.py:1323
↓ 2 callersMethod_make_layer
A layer is a combination of inverted residual blocks
yolact_edge/backbone.py:549
↓ 2 callersMethod_update_str
(self)
yolact_edge/utils/functions.py:71
↓ 2 callersMethodadd_bbox
Note that bbox should be a list or tuple of (x1, y1, x2, y2)
eval.py:296
↓ 2 callersMethodadd_images
(self, key, value)
yolact_edge/utils/tensorboard_helper.py:19
↓ 2 callersMethodadd_mask
The segmentation should be the full mask, the size of the image and with size [h, w].
eval.py:310
↓ 2 callersMethodadd_text
(self, key, value)
yolact_edge/utils/tensorboard_helper.py:15
↓ 2 callersFunctionbbox_iou
(bbox1, bbox2, iscrowd=False)
eval.py:386
↓ 2 callersFunctionbuild_batch_data_sampler
(sampler, images_per_batch)
yolact_edge/data/sampler_utils.py:41
↓ 2 callersFunctioncompute_hits
(bboxes, anchors, iou_threshold=0.5)
yolact_edge/scripts/optimize_bboxes.py:104
↓ 2 callersFunctionconstruct_backbone
Constructs a backbone given a backbone config object (see config.py).
yolact_edge/backbone.py:600
↓ 2 callersFunctionconvert_to_tensorrt
(net, cfg, args, transform)
yolact_edge/utils/tensorrt.py:9
↓ 2 callersFunctioncrop
"Crop" predicted masks by zeroing out everything not in the predicted bbox. Vectorized by Chong (thanks Chong). Args: - masks sh
yolact_edge/layers/box_utils.py:305
↓ 2 callersMethoddetect
Perform nms for only the max scoring class that isn't background (class 0)
yolact_edge/layers/functions/detection.py:85
↓ 2 callersMethodfreeze_bn
Adapted from https://discuss.pytorch.org/t/how-to-train-with-frozen-batchnorm/12106/8
yolact_edge/yolact.py:1335
↓ 2 callersFunctionget_coco_cat
transformed_cat_id is [0,80) as indices in cfg.dataset.class_names
eval.py:281
↓ 2 callersMethodget_interrupt
(save_folder)
yolact_edge/utils/functions.py:134
↓ 2 callersFunctionget_label_map
()
yolact_edge/data/coco.py:13
↓ 2 callersFunctionget_label_map
()
yolact_edge/data/youtube_vis.py:17
↓ 2 callersMethodinit_weights
Initialize weights for training.
yolact_edge/yolact.py:1266
↓ 2 callersFunctionmake_priors
(conv_size, scales, aspect_ratios)
yolact_edge/scripts/optimize_bboxes.py:74
↓ 2 callersMethodmake_priors
Note that priors are [x,y,width,height] where (x,y) is the center of the box.
yolact_edge/yolact.py:262
↓ 2 callersFunctionmask_iou
Inputs inputs are matricies of size _ x N. Output is size _1 x _2. Note: if iscrowd is True, then mask2 should be the crowd.
eval.py:366
↓ 2 callersFunctionpaint_mask
(img_numpy, mask, color)
yolact_edge/scripts/compute_masks.py:22
↓ 2 callersFunctionpoint_form
Convert prior_boxes to (xmin, ymin, xmax, ymax) representation for comparison to point form ground truth data. Args: boxes: (tensor)
yolact_edge/layers/box_utils.py:7
↓ 2 callersFunctionprep_benchmark
(dets_out, h, w)
eval.py:262
↓ 2 callersFunctionprep_coco_cats
Prepare inverted table for category id lookup given a coco cats object.
eval.py:273
↓ 2 callersFunctionprep_metrics
Returns a list of APs for this image, with each element being for a class
eval.py:391
↓ 2 callersMethodpull_frame
(self, vid_name, frame_annot_id, prev_frame_annot_id, file_name, target, num_crowds, require_seeds=False, seed
yolact_edge/data/youtube_vis.py:332
↓ 2 callersMethodpull_item
Args: index (int): Index Returns: tuple: Tuple (image, target, masks, height, width, crowd).
yolact_edge/data/coco.py:132
↓ 2 callersMethodpull_video
Args: index (int): Index Returns: tuple: Tuple (image, target, masks, height, width, crowd).
yolact_edge/data/youtube_vis.py:140
↓ 2 callersMethodpush
(self, score:float, is_true:bool)
eval.py:515
↓ 2 callersMethodreadImage
(name)
yolact_edge/data/flying_chairs.py:107
↓ 2 callersMethodset_step
(self, step)
yolact_edge/utils/tensorboard_helper.py:23
↓ 2 callersFunctionsetup_eval
()
train.py:698
↓ 2 callersFunctionsetup_logger
(name="yolact", output=None, distributed_rank=0, abbrev_name=None, logging_level=logging.DEBUG)
yolact_edge/utils/logging_helper.py:39
↓ 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, t
yolact_edge/utils/timer.py:35
↓ 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_stac
yolact_edge/utils/timer.py:55
↓ 2 callersMethodtarget_in_frame
(target, frame_id, true_on_reference=False)
yolact_edge/data/youtube_vis.py:323
↓ 2 callersFunctionundo_image_transformation
Takes a transformed image tensor and returns a numpy ndarray that is untransformed. Arguments w and h are the original height and width of th
yolact_edge/layers/output_utils.py:142
↓ 1 callersMethod__init__
(self, width, height, mean=MEANS, pad_gt=True)
yolact_edge/utils/augmentations.py:141
↓ 1 callersMethod__len__
(self)
yolact_edge/data/coco.py:129
↓ 1 callersMethod__len__
(self)
yolact_edge/data/youtube_vis.py:319
↓ 1 callersMethod_make_layer
Each layer is a sequence of conv layers usually preceded by a max pooling. Adapted from torchvision.models.vgg.make_layers.
yolact_edge/backbone.py:348
↓ 1 callersFunctionadd
(val)
yolact_edge/scripts/make_grid.py:113
↓ 1 callersMethodadd_gt_positives
Call this once per image.
eval.py:518
↓ 1 callersMethodadd_layer
Add a downsample layer to the backbone as per what SSD does.
yolact_edge/backbone.py:417
↓ 1 callersFunctionaugment_boxes
(bboxes)
yolact_edge/scripts/augment_bbox.py:18
↓ 1 callersFunctionbadhash
Just a quick and dirty hash function for doing a deterministic shuffle based on image_id. Source: https://stackoverflow.com/questions/66
eval.py:575
next →1–100 of 396, ranked by callers