Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/AILab-CVC/YOLO-World
/ types & classes
Types & classes
69 in github.com/AILab-CVC/YOLO-World
⨍
Functions
283
◇
Types & classes
69
↓ 2 callers
Class
BNContrastiveHead
Batch Norm Contrastive Head for YOLO-World using batch norm instead of l2-normalization Args: embed_dims (int): embed dim of text and
yolo_world/models/dense_heads/yolo_world_head.py:68
↓ 2 callers
Class
ContrastiveHead
Contrastive Head for YOLO-World compute the region-text scores according to the similarity between image and text features Args: e
yolo_world/models/dense_heads/yolo_world_head.py:28
↓ 2 callers
Class
DeployModel
deploy/easydeploy/model/model.py:24
↓ 2 callers
Class
MMYOLOBackend
deploy/easydeploy/model/backend.py:7
↓ 1 callers
Class
Decoder
deploy/easydeploy/examples/numpy_coder.py:18
↓ 1 callers
Class
DeployFocus
deploy/easydeploy/backbone/focus.py:8
↓ 1 callers
Class
EngineBuilder
deploy/easydeploy/tools/build_engine.py:17
↓ 1 callers
Class
GConvFocus
deploy/easydeploy/backbone/focus.py:59
↓ 1 callers
Class
ImagePromptEncoder
yolo_world/models/detectors/yolo_world_image.py:15
↓ 1 callers
Class
LabelAnnotator
deploy/tflite_demo.py:18
↓ 1 callers
Class
LabelAnnotator
deploy/onnx_demo.py:22
↓ 1 callers
Class
LabelAnnotator
demo/image_demo.py:21
↓ 1 callers
Class
LabelAnnotator
demo/image_prompt_demo.py:32
↓ 1 callers
Class
LabelAnnotator
demo/gradio_demo.py:32
↓ 1 callers
Class
MaxSigmoidAttnBlock
Max Sigmoid attention block.
yolo_world/models/layers/yolo_bricks.py:16
↓ 1 callers
Class
ModelType
deploy/easydeploy/examples/config.py:10
↓ 1 callers
Class
NcnnFocus
deploy/easydeploy/backbone/focus.py:26
↓ 1 callers
Class
ORTWrapper
deploy/easydeploy/model/backendwrapper.py:140
↓ 1 callers
Class
PreProcess
deploy/easydeploy/tools/image-demo.py:43
↓ 1 callers
Class
Preprocess
deploy/easydeploy/examples/preprocess.py:9
↓ 1 callers
Class
RepBNContrastiveHead
Batch Norm Contrastive Head for YOLO-World using batch norm instead of l2-normalization Args: embed_dims (int): embed dim of text and
yolo_world/models/dense_heads/yolo_world_head.py:110
↓ 1 callers
Class
RepConvMaxSigmoidAttnBlock
Max Sigmoid attention block.
yolo_world/models/layers/yolo_bricks.py:178
↓ 1 callers
Class
RepMatrixMaxSigmoidAttnBlock
Max Sigmoid attention block.
yolo_world/models/layers/yolo_bricks.py:101
↓ 1 callers
Class
TRTWrapper
deploy/easydeploy/model/backendwrapper.py:19
↓ 1 callers
Class
VanillaSigmoidBlock
Sigmoid attention block.
yolo_world/models/layers/yolo_bricks.py:506
Class
BaseMultiModalMixImageTransform
A Base Transform of Multimodal multiple images mixed. Suitable for training on multiple images mixed data augmentation like mosaic and mixup.
yolo_world/datasets/transformers/mm_mix_img_transforms.py:17
Class
CoVMSELoss
yolo_world/models/losses/dynamic_loss.py:12
Class
DeployC2f
deploy/easydeploy/backbone/common.py:6
Class
EfficientCSPLayerWithTwoConv
Sigmoid-attention based CSP layer with two convolution layers.
yolo_world/models/layers/yolo_bricks.py:551
Class
HuggingCLIPLanguageBackbone
yolo_world/models/backbones/mm_backbone.py:59
Class
HuggingVisionBackbone
yolo_world/models/backbones/mm_backbone.py:15
Class
ImagePoolingAttentionModule
yolo_world/models/layers/yolo_bricks.py:427
Class
LoadText
yolo_world/datasets/transformers/mm_transforms.py:101
Class
MaxSigmoidCSPLayerWithTwoConv
Sigmoid-attention based CSP layer with two convolution layers.
yolo_world/models/layers/yolo_bricks.py:257
Class
MultiModalDataset
Multi-modal dataset.
yolo_world/datasets/mm_dataset.py:14
Class
MultiModalMixedDataset
Multi-modal Mixed dataset. mix "detection dataset" and "caption dataset" Args: dataset_type (str): dataset type, 'detection' or 'capti
yolo_world/datasets/mm_dataset.py:94
Class
MultiModalMosaic
Mosaic augmentation. Given 4 images, mosaic transform combines them into one output image. The output image is composed of the parts from eac
yolo_world/datasets/transformers/mm_mix_img_transforms.py:207
Class
MultiModalMosaic9
Mosaic9 augmentation. Given 9 images, mosaic transform combines them into one output image. The output image is composed of the parts from ea
yolo_world/datasets/transformers/mm_mix_img_transforms.py:514
Class
MultiModalYOLOBackbone
yolo_world/models/backbones/mm_backbone.py:192
Class
ONNXNMSop
deploy/easydeploy/nms/ort_nms.py:100
Class
PseudoLanguageBackbone
Pseudo Language Backbone Args: text_embed_path (str): path to the text embedding file
yolo_world/models/backbones/mm_backbone.py:141
Class
RandomLoadText
yolo_world/datasets/transformers/mm_transforms.py:11
Class
RepConvMaxSigmoidCSPLayerWithTwoConv
Sigmoid-attention based CSP layer with two convolution layers.
yolo_world/models/layers/yolo_bricks.py:370
Class
RepMaxSigmoidCSPLayerWithTwoConv
Sigmoid-attention based CSP layer with two convolution layers.
yolo_world/models/layers/yolo_bricks.py:313
Class
RepYOLOWorldHeadModule
yolo_world/models/dense_heads/yolo_world_head.py:293
Class
SimpleYOLOWorldDetector
Implementation of YOLO World Series
yolo_world/models/detectors/yolo_world.py:109
Class
TASK_TYPE
deploy/easydeploy/examples/config.py:4
Class
TRTEfficientNMSop
deploy/easydeploy/nms/trt_nms.py:10
Class
TRTbatchedNMSop
TensorRT NMS operation.
deploy/easydeploy/nms/trt_nms.py:61
Class
V3DetDataset
Objects365 v1 dataset for detection.
yolo_world/datasets/yolov5_v3det.py:34
Class
YOLOWDetDataPreprocessor
Rewrite collate_fn to get faster training speed. Note: It must be used together with `mmyolo.datasets.utils.yolow_collate`
yolo_world/models/data_preprocessors/data_preprocessor.py:15
Class
YOLOWorldDetector
Implementation of YOLOW Series
yolo_world/models/detectors/yolo_world.py:12
Class
YOLOWorldDualPAFPN
Path Aggregation Network used in YOLO World v8.
yolo_world/models/necks/yolo_world_pafpn.py:150
Class
YOLOWorldHead
YOLO-World Head
yolo_world/models/dense_heads/yolo_world_head.py:348
Class
YOLOWorldHeadModule
Head Module for YOLO-World Args: embed_dims (int): embed dim for text feautures and image features use_bn_head (bool): use batch
yolo_world/models/dense_heads/yolo_world_head.py:135
Class
YOLOWorldImageDetector
Implementation of YOLO World Series
yolo_world/models/detectors/yolo_world_image.py:164
Class
YOLOWorldPAFPN
Path Aggregation Network used in YOLO World Following YOLOv8 PAFPN, including text to image fusion
yolo_world/models/necks/yolo_world_pafpn.py:16
Class
YOLOWorldSegAssigner
yolo_world/models/assigner/task_aligned_assigner.py:9
Class
YOLOWorldSegHead
yolo_world/models/dense_heads/yolo_world_seg_head.py:216
Class
YOLOWorldSegHeadModule
yolo_world/models/dense_heads/yolo_world_seg_head.py:30
Class
YOLOWv5OptimizerConstructor
YOLO World v5 constructor for optimizers.
yolo_world/engine/optimizers/yolow_v5_optim_constructor.py:19
Class
YOLOXMultiModalMixUp
MixUp data augmentation for YOLOX. .. code:: text mixup transform +---------------+--------------+
yolo_world/datasets/transformers/mm_mix_img_transforms.py:941
Class
YOLOv5GeneralGroundingDataset
Mixed grounding dataset.
yolo_world/datasets/yolov5_cc3m_grounding.py:13
Class
YOLOv5LVISV1Dataset
Dataset for YOLOv5 LVIS Dataset. We only add `BatchShapePolicy` function compared with Objects365V1Dataset. See `mmyolo/datasets/utils.py#Bat
yolo_world/datasets/yolov5_lvis.py:9
Class
YOLOv5MixedGroundingDataset
Mixed grounding dataset.
yolo_world/datasets/yolov5_mixed_grounding.py:13
Class
YOLOv5MultiModalMixUp
MixUp data augmentation for YOLOv5. .. code:: text The mixup transform steps are as follows: 1. Another random image is picked by d
yolo_world/datasets/transformers/mm_mix_img_transforms.py:809
Class
YOLOv5Objects365V1Dataset
Dataset for YOLOv5 VOC Dataset. We only add `BatchShapePolicy` function compared with Objects365V1Dataset. See `mmyolo/datasets/utils.py#Batc
yolo_world/datasets/yolov5_obj365v1.py:9
Class
YOLOv5Objects365V2Dataset
Dataset for YOLOv5 VOC Dataset. We only add `BatchShapePolicy` function compared with Objects365V1Dataset. See `mmyolo/datasets/utils.py#Batc
yolo_world/datasets/yolov5_obj365v2.py:9
Class
YOLOv5V3DetDataset
Dataset for YOLOv5 VOC Dataset. We only add `BatchShapePolicy` function compared with Objects365V1Dataset. See `mmyolo/datasets/utils.py#Batc
yolo_world/datasets/yolov5_v3det.py:104