MCPcopy Index your code

hub / github.com/AILab-CVC/YOLO-World / functions

Functions283 in github.com/AILab-CVC/YOLO-World

Methodforward
Forward function. including multi-level image features, text features: BxLxD
yolo_world/models/necks/yolo_world_pafpn.py:106
Methodforward
Forward function.
yolo_world/models/necks/yolo_world_pafpn.py:196
Methodforward
(self, image: Tensor)
yolo_world/models/backbones/mm_backbone.py:31
Methodforward
(self, text: List[List[str]])
yolo_world/models/backbones/mm_backbone.py:86
Methodforward
(self, text: List[List[str]])
yolo_world/models/backbones/mm_backbone.py:166
Methodforward
(self, image: Tensor, text: List[List[str]])
yolo_world/models/backbones/mm_backbone.py:225
Methodforward
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_seg_head.py:173
Methodforward
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_seg_head.py:314
Methodforward
Forward function of contrastive learning.
yolo_world/models/dense_heads/yolo_world_head.py:46
Methodforward
Forward function of contrastive learning.
yolo_world/models/dense_heads/yolo_world_head.py:88
Methodforward
Forward function of contrastive learning.
yolo_world/models/dense_heads/yolo_world_head.py:127
Methodforward
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_head.py:246
Methodforward
(self, img_feats: Tuple[Tensor])
yolo_world/models/dense_heads/yolo_world_head.py:341
Methodforward
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_head.py:396
Methodforward
Perform normalization, padding and bgr2rgb conversion based on ``DetDataPreprocessorr``. Args: data (dict): Data sampled
yolo_world/models/data_preprocessors/data_preprocessor.py:24
Methodforward
Forward process.
yolo_world/models/layers/yolo_bricks.py:68
Methodforward
Forward process.
yolo_world/models/layers/yolo_bricks.py:150
Methodforward
Forward process.
yolo_world/models/layers/yolo_bricks.py:229
Methodforward
Forward process.
yolo_world/models/layers/yolo_bricks.py:303
Methodforward
Forward process.
yolo_world/models/layers/yolo_bricks.py:360
Methodforward
Forward process.
yolo_world/models/layers/yolo_bricks.py:417
Methodforward
(self, text_features, image_features)
yolo_world/models/layers/yolo_bricks.py:468
Methodforward
Forward process.
yolo_world/models/layers/yolo_bricks.py:542
Methodforward
Forward process.
yolo_world/models/layers/yolo_bricks.py:595
Methodforward
Forward function of loss.
yolo_world/models/losses/dynamic_loss.py:25
Methodforward
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
Methodforward_gvp
(x: Tensor)
deploy/easydeploy/model/model.py:216
Methodforward_single
(x: Tensor, convs: nn.Module)
deploy/easydeploy/model/model.py:197
Methodforward_single
Forward feature of a single scale level.
yolo_world/models/dense_heads/yolo_world_seg_head.py:187
Methodforward_single
Forward feature of a single scale level.
yolo_world/models/dense_heads/yolo_world_head.py:256
Methodforward_single
Forward features from the upstream network.
yolo_world/models/dense_heads/yolo_world_head.py:316
Methodforward_text
(self, text: List[List[str]])
yolo_world/models/backbones/mm_backbone.py:234
Methodforward_tokenizer
(self, texts)
yolo_world/models/backbones/mm_backbone.py:79
Methodget_data_info
Get annotation by index.
yolo_world/datasets/mm_dataset.py:115
Methodget_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
Methodget_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
Methodget_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
Methodget_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
Functioninference
(ort_session, image_path, texts, output_dir, size=(640
deploy/onnx_demo.py:90
Functioninference_with_postprocessing
(ort_session, image_path, texts,
deploy/onnx_demo.py:123
Methodinit_weights
Initialize the weight and bias of PPYOLOE head.
yolo_world/models/dense_heads/yolo_world_seg_head.py:48
Methodinit_weights
Initialize the weight and bias of PPYOLOE head.
yolo_world/models/dense_heads/yolo_world_head.py:155
Methodload_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
Methodload_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
Methodload_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
Methodloss
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
Methodloss
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
Methodloss
Calculate losses from a batch of inputs and data samples.
yolo_world/models/detectors/yolo_world.py:157
Methodloss
Calculate losses from a batch of inputs and data samples.
yolo_world/models/detectors/yolo_world_image.py:185
Methodloss_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
Methodloss_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
Methodmetainfo
(self)
yolo_world/datasets/mm_dataset.py:51
Methodmix_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
Methodmix_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
Methodmix_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
Methodmix_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
Functiononnx_nms
( boxes: torch.Tensor, scores: torch.Tensor, max_output_boxes_per_class: int = 100, iou_thresh
deploy/easydeploy/nms/ort_nms.py:189
Methodpredict
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
Methodpredict
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
Methodpredict
Predict results from a batch of inputs and data samples with post- processing.
yolo_world/models/detectors/yolo_world.py:170
Methodpredict
Predict results from a batch of inputs and data samples with post- processing.
yolo_world/models/detectors/yolo_world_image.py:195
Methodresolve_text_background_xyxy
( center_coordinates, text_wh, position, )
deploy/tflite_demo.py:21
Methodresolve_text_background_xyxy
( center_coordinates, text_wh, position, )
deploy/onnx_demo.py:25
Methodresolve_text_background_xyxy
( center_coordinates, text_wh, position, )
demo/image_demo.py:24
Methodresolve_text_background_xyxy
( center_coordinates, text_wh, position, )
demo/image_prompt_demo.py:35
Methodresolve_text_background_xyxy
( center_coordinates, text_wh, position, )
demo/gradio_demo.py:35
Functionrtmdet_bbox_decoder
(priors: Tensor, bbox_preds: Tensor, stride: Optional[Tensor])
deploy/easydeploy/bbox_code/bbox_coder.py:28
Functionrun_image
(runner, vision_encoder, vision_processor, padding_token,
demo/image_prompt_demo.py:90
Functionrun_image
(runner, image, text, max_num_boxes, score_thr,
demo/gradio_demo.py:72
Methodscale_bbox
(bbox, scale, image_width, image_height)
yolo_world/models/detectors/yolo_world_image.py:54
Methodspecial_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
Methodsymbolic
( g, boxes: Tensor, scores: Tensor, max_output_boxes_per_class
deploy/easydeploy/nms/ort_nms.py:171
Methodsymbolic
(g, boxes: Tensor, scores: Tensor, background_class: int =
deploy/easydeploy/nms/trt_nms.py:35
Methodsymbolic
( g, boxes: Tensor, scores: Tensor, plugin_version: str = '1', shareLo
deploy/easydeploy/nms/trt_nms.py:92
Methodtrain
(self, mode=True)
yolo_world/models/backbones/mm_backbone.py:135
Methodtrain
Convert the model into training mode while keep normalization layer frozen.
yolo_world/models/backbones/mm_backbone.py:219
Methodtrain
Convert the model into training mode while keep normalization layer frozen.
yolo_world/models/dense_heads/yolo_world_seg_head.py:166
Methodtrain
(self, mode=True)
yolo_world/models/dense_heads/yolo_world_head.py:241
Methodtrain
(self, mode=True)
yolo_world/models/detectors/yolo_world_image.py:48
Methodtransform
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
Functionyolov5_bbox_decoder
(priors: Tensor, bbox_preds: Tensor, stride: Tensor)
deploy/easydeploy/bbox_code/bbox_coder.py:8
Functionyolow_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
Functionyolox_bbox_decoder
(priors: Tensor, bbox_preds: Tensor, stride: Optional[Tensor])
deploy/easydeploy/bbox_code/bbox_coder.py:40
← previous201–283 of 283, ranked by callers