MCPcopy Create free account

hub / github.com/WisconsinAIVision/yolact_edge / functions

Functions396 in github.com/WisconsinAIVision/yolact_edge

Method__str__
(self)
yolact_edge/utils/functions.py:40
Method__str__
(self)
yolact_edge/utils/functions.py:81
Methodadd_layer
Add a downsample layer to the backbone as per what SSD does.
yolact_edge/backbone.py:147
Methodadd_layer
Add a downsample layer to the backbone as per what SSD does.
yolact_edge/backbone.py:300
Methodadd_layer
TODO: Need to make sure that this works as intended.
yolact_edge/backbone.py:576
Functionadd_randomize
(val)
yolact_edge/scripts/make_grid.py:123
Methodall_in
(x, y)
yolact_edge/yolact.py:1274
Functionbarrier
()
yolact_edge/utils/misc.py:29
Methodcoefficient_nms
(self, coeffs, scores, cos_threshold=0.9, top_k=400)
yolact_edge/layers/functions/detection.py:112
Functioncollate_fn_flying_chairs
(batch)
yolact_edge/data/flying_chairs.py:17
Functioncollate_fn_youtube_vis
(batch)
yolact_edge/data/youtube_vis.py:28
Functioncollate_fn_youtube_vis_eval
(batch)
yolact_edge/data/youtube_vis.py:24
Functioncompute_validation_loss
(net, data_loader, criterion)
train.py:658
Functionconv
(batchNorm, in_planes, out_planes, kernel_size=3, stride=1)
yolact_edge/yolact.py:402
Functionconv_bn
(in_features, out_features, kernel_size=3, stride=1, dilation=1)
yolact_edge/yolact.py:456
Functionconv_bn_lrelu
(in_features, out_features, kernel_size=3, stride=1, padding=1, dilation=1, batch_norm=True)
yolact_edge/yolact.py:379
Functionconv_bn_relu
(in_features, out_features, kernel_size=3, stride=1, dilation=1)
yolact_edge/yolact.py:427
Functionconv_only
(in_features, out_features, kernel_size=3, stride=1, dilation=1, groups=1)
yolact_edge/yolact.py:443
Functionconv_relu
(in_features, out_features, kernel_size=3, stride=1, dilation=1, groups=1)
yolact_edge/yolact.py:435
Functiondeconv
(in_features, out_features)
yolact_edge/yolact.py:416
Functiondetection_collate
Custom collate fn for dealing with batches of images that have a different number of associated object annotations (bounding boxes). Argument
yolact_edge/data/__init__.py:10
Functiondisable
Disables the given function name fom being considered for the average or outputted in print_stats.
yolact_edge/utils/timer.py:19
Functiondisable_all
()
yolact_edge/utils/timer.py:11
Functiondo_nothing
(img=None, masks=None, boxes=None, labels=None, seeds=None, require_seeds=False)
yolact_edge/utils/augmentations.py:886
Functionenable
Enables function names disabled by disable.
yolact_edge/utils/timer.py:23
Functionenable_all
()
yolact_edge/utils/timer.py:15
Functionexport
(val)
yolact_edge/scripts/make_grid.py:127
Methodfix_shape
(classes, boxes, masks, scores)
yolact_edge/layers/functions/detection.py:195
Methodflatten_index_select
(x, idx, end_dim=None)
yolact_edge/layers/functions/detection.py:205
MethodformatMessage
(self, record)
yolact_edge/utils/logging_helper.py:18
Methodforward
(self, x)
yolact_edge/yolact.py:52
Methodforward
Args: - x: The convOut from a layer in the backbone network Size: [batch_size, in_channels, conv_h, conv_w])
yolact_edge/yolact.py:190
Methodforward
Args: - x: The convOut from a layer in the backbone network Size: [batch_size, in_channels, conv_h, conv_w])
yolact_edge/yolact.py:314
Methodforward
(self, a, b)
yolact_edge/yolact.py:475
Methodforward
(self, a, b)
yolact_edge/yolact.py:481
Methodforward
(self, a, b)
yolact_edge/yolact.py:492
Methodforward
(self, a, b)
yolact_edge/yolact.py:505
Methodforward
(self, preds)
yolact_edge/yolact.py:515
Methodforward
(self, a, b)
yolact_edge/yolact.py:536
Methodforward
(self, x)
yolact_edge/yolact.py:572
Methodforward
(self, x)
yolact_edge/yolact.py:604
Methodforward
(self, x)
yolact_edge/yolact.py:642
Methodforward
(self, x)
yolact_edge/yolact.py:661
Methodforward
(self, target_feat, source_feat)
yolact_edge/yolact.py:706
Methodforward
(self, concat0)
yolact_edge/yolact.py:781
Methodforward
(self, c3, f2, f3)
yolact_edge/yolact.py:836
Methodforward
Args: - convouts (list): A list of convouts for the corresponding layers in in_channels. Returns: - A list of
yolact_edge/yolact.py:873
Methodforward
Args: - convouts (list): A list of convouts for the corresponding layers in in_channels. Returns: - A list of
yolact_edge/yolact.py:943
Methodforward
Args: - convouts (list): A list of convouts for the corresponding layers in in_channels. Returns: - A list of
yolact_edge/yolact.py:1026
Methodforward
The input should be of size [batch_size, 3, img_h, img_w]
yolact_edge/yolact.py:1611
Methodforward
(self, x)
yolact_edge/backbone.py:30
Methodforward
Returns a list of convouts for each layer.
yolact_edge/backbone.py:115
Methodforward
(self, x)
yolact_edge/backbone.py:235
Methodforward
Returns a list of convouts for each layer.
yolact_edge/backbone.py:288
Methodforward
Returns a list of convouts for each layer.
yolact_edge/backbone.py:389
Methodforward
(self, x)
yolact_edge/backbone.py:496
Methodforward
Returns a list of convouts for each layer.
yolact_edge/backbone.py:563
Methodforward
(self, img)
yolact_edge/utils/augmentations.py:857
Methodforward
(self, x)
yolact_edge/layers/interpolate.py:16
Methodforward
(self, preds, gt)
yolact_edge/layers/modules/optical_flow_loss.py:12
Methodforward
Multibox Loss Args: predictions (tuple): A tuple containing loc preds, conf preds, mask preds, and prior boxes from SS
yolact_edge/layers/modules/multibox_loss.py:46
Functionforward_hook
(self, inputs, outputs)
yolact_edge/utils/tensorrt.py:90
Methodget_color
(j, on_gpu=None)
yolact_edge/inference.py:208
Functionget_world_size
()
yolact_edge/utils/misc.py:13
Functionindex2d
Indexes a tensor by a 2d index. In effect, this does out[i, j] = src[i, idx[i, j]] Both src and idx should have the same si
yolact_edge/layers/box_utils.py:332
Methodinit_backbone
Initializes the backbone weights for training.
yolact_edge/backbone.py:132
Methodinit_backbone
The path here comes from detectron. So we load it differently.
yolact_edge/backbone.py:157
Methodinit_backbone
Initializes the backbone weights for training.
yolact_edge/backbone.py:304
Methodinit_backbone
Initializes the backbone weights for training.
yolact_edge/backbone.py:582
Methodis_finished
(self)
yolact_edge/utils/functions.py:68
Functionis_main_process
()
yolact_edge/utils/misc.py:25
Functionlog_once
(obj, msg_key, name, message, level=logging.WARNING, *args, **kwargs)
yolact_edge/utils/logging_helper.py:30
Functionlogit
(x)
yolact_edge/scripts/compute_masks.py:41
Methodmake_extra
(num_layers)
yolact_edge/yolact.py:169
Functionplay_video
()
eval.py:724
Functionpredict_flow
(in_features)
yolact_edge/yolact.py:423
Functionprep_frame
(inp)
eval.py:715
Methodprint
(self)
yolact_edge/data/config.py:127
Functionprint_stats
Prints the current timing information into a table.
yolact_edge/utils/timer.py:85
Methodpull_anno
Returns the original annotation of image at index Note: not using self.__getitem__(), as any transformations passed in could mess up
yolact_edge/data/coco.py:232
Methodpull_anno
Returns the original annotation of image at index Note: not using self.__getitem__(), as any transformations passed in could mess up
yolact_edge/data/youtube_vis.py:435
Methodpull_image
Returns the original image object at index in PIL form Note: not using self.__getitem__(), as any transformations passed in could mes
yolact_edge/data/coco.py:217
Methodpull_image
Returns the original image object at index in PIL form Note: not using self.__getitem__(), as any transformations passed in could mes
yolact_edge/data/youtube_vis.py:420
Functionreset
Resets the current timer. Call this at the start of an iteration.
yolact_edge/utils/timer.py:27
Functionset_x
(x, scales, aspect_ratios)
yolact_edge/scripts/optimize_bboxes.py:141
Functionsigmoid
(x)
yolact_edge/scripts/compute_masks.py:44
Functionstep
(x, x_func, bboxes, base_hits, optim_idx)
yolact_edge/scripts/optimize_bboxes.py:115
Functionstr2bool
(v)
eval.py:37
Functionstr2bool
(v)
train.py:33
Functionstr2bool
(v)
yolact_edge/inference.py:17
Functiontensor_as
(x: float, t: torch.Tensor)
yolact_edge/layers/warp_utils.py:40
Functionupdate_angle
(val)
yolact_edge/scripts/make_grid.py:69
Functionupdate_centerx
(val)
yolact_edge/scripts/make_grid.py:75
Functionupdate_centery
(val)
yolact_edge/scripts/make_grid.py:81
Functionupdate_scale
(val)
yolact_edge/scripts/make_grid.py:63
Functionupdate_spacing
(val)
yolact_edge/scripts/make_grid.py:87
← previous301–396 of 396, ranked by callers