MCPcopy Create free account

hub / github.com/WeChatCV/WeVisionOne / functions

Functions221 in github.com/WeChatCV/WeVisionOne

Functiongen_encoder_output_proposals
Input: - memory: bs, \sum{hw}, d_model - memory_padding_mask: bs, \sum{hw} - spatial_shapes: nlevel, 2 - learnedw
WeVisionOne/pixel_decoder/groundingdino/utils.py:55
Functiongeneralized_box_iou
Generalized IoU from https://giou.stanford.edu/ The boxes should be in [x0, y0, x1, y1] format Returns a [N, M] pairwise matrix, where
WeVisionOne/utils/box_ops.py:41
Functiongeneralized_box_iou_pairwise
Generalized IoU from https://giou.stanford.edu/ Input: - boxes1, boxes2: N,4 Output: - giou: N, 4
WeVisionOne/utils/box_ops.py:83
Functionis_dist_avail_and_initialized
()
WeVisionOne/utils/misc.py:107
Functionl1norm
L1-normalize columns of X
WeVisionOne/pixel_decoder/groundingdino/fuse_modules.py:36
Functionmasks_to_boxes
Compute the bounding boxes around the provided masks The masks should be in format [N, H, W] where N is the number of masks, (H, W) are the spatia
WeVisionOne/utils/misc.py:114
Functionmasks_to_boxes
Compute the bounding boxes around the provided masks The masks should be in format [N, H, W] where N is the number of masks, (H, W) are the spati
WeVisionOne/utils/box_ops.py:107
Functionms_deform_attn_backward
WeVisionOne/pixel_decoder/ops/src/ms_deform_attn.h:46
Functionms_deform_attn_cpu_backward
WeVisionOne/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp:34
Functionms_deform_attn_cpu_forward
WeVisionOne/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp:22
Functionms_deform_attn_forward
WeVisionOne/pixel_decoder/ops/src/ms_deform_attn.h:25
Functionnested_tensor_from_tensor_list
(tensor_list: List[Tensor])
WeVisionOne/utils/misc.py:49
Methodobtain_spatial_feature
(self, features, mask=None, txt_feats=None, rand_shape=None)
WeVisionOne/heads/maskdino_head.py:74
Functionon_feats_btn
(click_img0, click_img1)
Inference/gradio_demo.py:57
Functionon_submit_btn
(target_img, v_feats_path)
Inference/gradio_demo.py:100
Methodpadding_constraints
(self)
WeVisionOne/backbone/eva02/det/vit.py:588
Methodpadding_constraints
This property is a generalization of size_divisibility. Some backbones and training recipes require specific padding constraints, suc
WeVisionOne/backbone/eva02/det/backbone.py:45
Methodrm_self_attn_modules
(self)
WeVisionOne/transformer_decoder/dino_decoder.py:205
Methodrm_self_attn_modules
(self)
WeVisionOne/pixel_decoder/groundingdino/transformer.py:855
Methodsize_divisibility
Some backbones require the input height and width to be divisible by a specific integer. This is typically true for encoder / decoder
WeVisionOne/backbone/eva02/det/backbone.py:34
Methodupdate_iter
(self, losses)
WeVisionOne/utils/misc.py:169
← previous201–221 of 221, ranked by callers