Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/WeChatCV/WeVisionOne
/ functions
Functions
221 in github.com/WeChatCV/WeVisionOne
⨍
Functions
221
◇
Types & classes
52
Function
gen_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
Function
generalized_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
Function
generalized_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
Function
is_dist_avail_and_initialized
()
WeVisionOne/utils/misc.py:107
Function
l1norm
L1-normalize columns of X
WeVisionOne/pixel_decoder/groundingdino/fuse_modules.py:36
Function
masks_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
Function
masks_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
Function
ms_deform_attn_backward
WeVisionOne/pixel_decoder/ops/src/ms_deform_attn.h:46
Function
ms_deform_attn_cpu_backward
WeVisionOne/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp:34
Function
ms_deform_attn_cpu_forward
WeVisionOne/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp:22
Function
ms_deform_attn_forward
WeVisionOne/pixel_decoder/ops/src/ms_deform_attn.h:25
Function
nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
WeVisionOne/utils/misc.py:49
Method
obtain_spatial_feature
(self, features, mask=None, txt_feats=None, rand_shape=None)
WeVisionOne/heads/maskdino_head.py:74
Function
on_feats_btn
(click_img0, click_img1)
Inference/gradio_demo.py:57
Function
on_submit_btn
(target_img, v_feats_path)
Inference/gradio_demo.py:100
Method
padding_constraints
(self)
WeVisionOne/backbone/eva02/det/vit.py:588
Method
padding_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
Method
rm_self_attn_modules
(self)
WeVisionOne/transformer_decoder/dino_decoder.py:205
Method
rm_self_attn_modules
(self)
WeVisionOne/pixel_decoder/groundingdino/transformer.py:855
Method
size_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
Method
update_iter
(self, losses)
WeVisionOne/utils/misc.py:169
← previous
201–221 of 221, ranked by callers