Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/AILab-CVC/YOLO-World
/ functions
Functions
283 in github.com/AILab-CVC/YOLO-World
⨍
Functions
283
◇
Types & classes
69
Method
forward
Forward function. including multi-level image features, text features: BxLxD
yolo_world/models/necks/yolo_world_pafpn.py:106
Method
forward
Forward function.
yolo_world/models/necks/yolo_world_pafpn.py:196
Method
forward
(self, image: Tensor)
yolo_world/models/backbones/mm_backbone.py:31
Method
forward
(self, text: List[List[str]])
yolo_world/models/backbones/mm_backbone.py:86
Method
forward
(self, text: List[List[str]])
yolo_world/models/backbones/mm_backbone.py:166
Method
forward
(self, image: Tensor, text: List[List[str]])
yolo_world/models/backbones/mm_backbone.py:225
Method
forward
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_seg_head.py:173
Method
forward
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_seg_head.py:314
Method
forward
Forward function of contrastive learning.
yolo_world/models/dense_heads/yolo_world_head.py:46
Method
forward
Forward function of contrastive learning.
yolo_world/models/dense_heads/yolo_world_head.py:88
Method
forward
Forward function of contrastive learning.
yolo_world/models/dense_heads/yolo_world_head.py:127
Method
forward
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_head.py:246
Method
forward
(self, img_feats: Tuple[Tensor])
yolo_world/models/dense_heads/yolo_world_head.py:341
Method
forward
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_head.py:396
Method
forward
Perform normalization, padding and bgr2rgb conversion based on ``DetDataPreprocessorr``. Args: data (dict): Data sampled
yolo_world/models/data_preprocessors/data_preprocessor.py:24
Method
forward
Forward process.
yolo_world/models/layers/yolo_bricks.py:68
Method
forward
Forward process.
yolo_world/models/layers/yolo_bricks.py:150
Method
forward
Forward process.
yolo_world/models/layers/yolo_bricks.py:229
Method
forward
Forward process.
yolo_world/models/layers/yolo_bricks.py:303
Method
forward
Forward process.
yolo_world/models/layers/yolo_bricks.py:360
Method
forward
Forward process.
yolo_world/models/layers/yolo_bricks.py:417
Method
forward
(self, text_features, image_features)
yolo_world/models/layers/yolo_bricks.py:468
Method
forward
Forward process.
yolo_world/models/layers/yolo_bricks.py:542
Method
forward
Forward process.
yolo_world/models/layers/yolo_bricks.py:595
Method
forward
Forward function of loss.
yolo_world/models/losses/dynamic_loss.py:25
Method
forward
Assign gt to bboxes. The assignment is done in following steps 1. compute alignment metric between all bbox (bbox of all pyramid
yolo_world/models/assigner/task_aligned_assigner.py:21
Method
forward_gvp
(x: Tensor)
deploy/easydeploy/model/model.py:216
Method
forward_single
(x: Tensor, convs: nn.Module)
deploy/easydeploy/model/model.py:197
Method
forward_single
Forward feature of a single scale level.
yolo_world/models/dense_heads/yolo_world_seg_head.py:187
Method
forward_single
Forward feature of a single scale level.
yolo_world/models/dense_heads/yolo_world_head.py:256
Method
forward_single
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_head.py:316
Method
forward_text
(self, text: List[List[str]])
yolo_world/models/backbones/mm_backbone.py:234
Method
forward_tokenizer
(self, texts)
yolo_world/models/backbones/mm_backbone.py:79
Method
get_data_info
Get annotation by index.
yolo_world/datasets/mm_dataset.py:115
Method
get_indexes
Call function to collect indexes. Args: dataset (:obj:`Dataset` or list): The dataset or cached list. Returns:
yolo_world/datasets/transformers/mm_mix_img_transforms.py:316
Method
get_indexes
Call function to collect indexes. Args: dataset (:obj:`Dataset` or list): The dataset or cached list. Returns:
yolo_world/datasets/transformers/mm_mix_img_transforms.py:626
Method
get_indexes
Call function to collect indexes. Args: dataset (:obj:`Dataset` or list): The dataset or cached list. Returns:
yolo_world/datasets/transformers/mm_mix_img_transforms.py:883
Method
get_indexes
Call function to collect indexes. Args: dataset (:obj:`Dataset` or list): The dataset or cached list. Returns:
yolo_world/datasets/transformers/mm_mix_img_transforms.py:1043
Function
inference
(ort_session, image_path, texts, output_dir, size=(640
deploy/onnx_demo.py:90
Function
inference_with_postprocessing
(ort_session, image_path, texts,
deploy/onnx_demo.py:123
Method
init_weights
Initialize the weight and bias of PPYOLOE head.
yolo_world/models/dense_heads/yolo_world_seg_head.py:48
Method
init_weights
Initialize the weight and bias of PPYOLOE head.
yolo_world/models/dense_heads/yolo_world_head.py:155
Method
load_data_list
Load annotations from an annotation file named as ``self.ann_file`` Returns: List[dict]: A list of annotation.
yolo_world/datasets/yolov5_cc3m_grounding.py:20
Method
load_data_list
Load annotations from an annotation file named as ``self.ann_file`` Returns: List[dict]: A list of annotation.
yolo_world/datasets/yolov5_mixed_grounding.py:20
Method
load_data_list
Load annotations from an annotation file named as ``self.ann_file`` Returns: List[dict]: A list of annotation.
yolo_world/datasets/yolov5_v3det.py:43
Method
loss
Perform forward propagation and loss calculation of the detection head on the features of the upstream network.
yolo_world/models/dense_heads/yolo_world_seg_head.py:275
Method
loss
Perform forward propagation and loss calculation of the detection head on the features of the upstream network.
yolo_world/models/dense_heads/yolo_world_head.py:357
Method
loss
Calculate losses from a batch of inputs and data samples.
yolo_world/models/detectors/yolo_world.py:157
Method
loss
Calculate losses from a batch of inputs and data samples.
yolo_world/models/detectors/yolo_world_image.py:185
Method
loss_and_predict
Perform forward propagation of the head, then calculate loss and predictions from the features and data samples.
yolo_world/models/dense_heads/yolo_world_seg_head.py:289
Method
loss_and_predict
Perform forward propagation of the head, then calculate loss and predictions from the features and data samples.
yolo_world/models/dense_heads/yolo_world_head.py:370
Method
metainfo
(self)
yolo_world/datasets/mm_dataset.py:51
Method
mix_img_transform
Mixed image data transformation. Args: results (dict): Result dict. Returns: results (dict): Updated result
yolo_world/datasets/transformers/mm_mix_img_transforms.py:328
Method
mix_img_transform
Mixed image data transformation. Args: results (dict): Result dict. Returns: results (dict): Updated result
yolo_world/datasets/transformers/mm_mix_img_transforms.py:638
Method
mix_img_transform
YOLOv5 MixUp transform function. Args: results (dict): Result dict Returns: results (dict): Updated result d
yolo_world/datasets/transformers/mm_mix_img_transforms.py:894
Method
mix_img_transform
YOLOX MixUp transform function. Args: results (dict): Result dict. Returns: results (dict): Updated result d
yolo_world/datasets/transformers/mm_mix_img_transforms.py:1054
Function
onnx_nms
( boxes: torch.Tensor, scores: torch.Tensor, max_output_boxes_per_class: int = 100, iou_thresh
deploy/easydeploy/nms/ort_nms.py:189
Method
predict
Perform forward propagation of the detection head and predict detection results on the features of the upstream network.
yolo_world/models/dense_heads/yolo_world_seg_head.py:319
Method
predict
Perform forward propagation of the detection head and predict detection results on the features of the upstream network.
yolo_world/models/dense_heads/yolo_world_head.py:401
Method
predict
Predict results from a batch of inputs and data samples with post- processing.
yolo_world/models/detectors/yolo_world.py:170
Method
predict
Predict results from a batch of inputs and data samples with post- processing.
yolo_world/models/detectors/yolo_world_image.py:195
Method
resolve_text_background_xyxy
( center_coordinates, text_wh, position, )
deploy/tflite_demo.py:21
Method
resolve_text_background_xyxy
( center_coordinates, text_wh, position, )
deploy/onnx_demo.py:25
Method
resolve_text_background_xyxy
( center_coordinates, text_wh, position, )
demo/image_demo.py:24
Method
resolve_text_background_xyxy
( center_coordinates, text_wh, position, )
demo/image_prompt_demo.py:35
Method
resolve_text_background_xyxy
( center_coordinates, text_wh, position, )
demo/gradio_demo.py:35
Function
rtmdet_bbox_decoder
(priors: Tensor, bbox_preds: Tensor, stride: Optional[Tensor])
deploy/easydeploy/bbox_code/bbox_coder.py:28
Function
run_image
(runner, vision_encoder, vision_processor, padding_token,
demo/image_prompt_demo.py:90
Function
run_image
(runner, image, text, max_num_boxes, score_thr,
demo/gradio_demo.py:72
Method
scale_bbox
(bbox, scale, image_width, image_height)
yolo_world/models/detectors/yolo_world_image.py:54
Method
special_init
Since YOLO series algorithms will inherit from YOLOv5Head, but different algorithms have special initialization process. The special_
yolo_world/models/dense_heads/yolo_world_seg_head.py:259
Method
symbolic
( g, boxes: Tensor, scores: Tensor, max_output_boxes_per_class
deploy/easydeploy/nms/ort_nms.py:171
Method
symbolic
(g, boxes: Tensor, scores: Tensor, background_class: int =
deploy/easydeploy/nms/trt_nms.py:35
Method
symbolic
( g, boxes: Tensor, scores: Tensor, plugin_version: str = '1', shareLo
deploy/easydeploy/nms/trt_nms.py:92
Method
train
(self, mode=True)
yolo_world/models/backbones/mm_backbone.py:135
Method
train
Convert the model into training mode while keep normalization layer frozen.
yolo_world/models/backbones/mm_backbone.py:219
Method
train
Convert the model into training mode while keep normalization layer frozen.
yolo_world/models/dense_heads/yolo_world_seg_head.py:166
Method
train
(self, mode=True)
yolo_world/models/dense_heads/yolo_world_head.py:241
Method
train
(self, mode=True)
yolo_world/models/detectors/yolo_world_image.py:48
Method
transform
Data augmentation function. The transform steps are as follows: 1. Randomly generate index list of other images. 2. Before Mo
yolo_world/datasets/transformers/mm_mix_img_transforms.py:110
Function
yolov5_bbox_decoder
(priors: Tensor, bbox_preds: Tensor, stride: Tensor)
deploy/easydeploy/bbox_code/bbox_coder.py:8
Function
yolow_collate
Rewrite collate_fn to get faster training speed. Args: data_batch (Sequence): Batch of data. use_ms_training (bool): Whether to use
yolo_world/datasets/utils.py:9
Function
yolox_bbox_decoder
(priors: Tensor, bbox_preds: Tensor, stride: Optional[Tensor])
deploy/easydeploy/bbox_code/bbox_coder.py:40
← previous
201–283 of 283, ranked by callers