MCPcopy Index your code

hub / github.com/AILab-CVC/YOLO-World / types & classes

Types & classes69 in github.com/AILab-CVC/YOLO-World

↓ 2 callersClassBNContrastiveHead
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 callersClassContrastiveHead
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 callersClassDeployModel
deploy/easydeploy/model/model.py:24
↓ 2 callersClassMMYOLOBackend
deploy/easydeploy/model/backend.py:7
↓ 1 callersClassDecoder
deploy/easydeploy/examples/numpy_coder.py:18
↓ 1 callersClassDeployFocus
deploy/easydeploy/backbone/focus.py:8
↓ 1 callersClassEngineBuilder
deploy/easydeploy/tools/build_engine.py:17
↓ 1 callersClassGConvFocus
deploy/easydeploy/backbone/focus.py:59
↓ 1 callersClassImagePromptEncoder
yolo_world/models/detectors/yolo_world_image.py:15
↓ 1 callersClassLabelAnnotator
deploy/tflite_demo.py:18
↓ 1 callersClassLabelAnnotator
deploy/onnx_demo.py:22
↓ 1 callersClassLabelAnnotator
demo/image_demo.py:21
↓ 1 callersClassLabelAnnotator
demo/image_prompt_demo.py:32
↓ 1 callersClassLabelAnnotator
demo/gradio_demo.py:32
↓ 1 callersClassMaxSigmoidAttnBlock
Max Sigmoid attention block.
yolo_world/models/layers/yolo_bricks.py:16
↓ 1 callersClassModelType
deploy/easydeploy/examples/config.py:10
↓ 1 callersClassNcnnFocus
deploy/easydeploy/backbone/focus.py:26
↓ 1 callersClassORTWrapper
deploy/easydeploy/model/backendwrapper.py:140
↓ 1 callersClassPreProcess
deploy/easydeploy/tools/image-demo.py:43
↓ 1 callersClassPreprocess
deploy/easydeploy/examples/preprocess.py:9
↓ 1 callersClassRepBNContrastiveHead
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 callersClassRepConvMaxSigmoidAttnBlock
Max Sigmoid attention block.
yolo_world/models/layers/yolo_bricks.py:178
↓ 1 callersClassRepMatrixMaxSigmoidAttnBlock
Max Sigmoid attention block.
yolo_world/models/layers/yolo_bricks.py:101
↓ 1 callersClassTRTWrapper
deploy/easydeploy/model/backendwrapper.py:19
↓ 1 callersClassVanillaSigmoidBlock
Sigmoid attention block.
yolo_world/models/layers/yolo_bricks.py:506
ClassBaseMultiModalMixImageTransform
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
ClassCoVMSELoss
yolo_world/models/losses/dynamic_loss.py:12
ClassDeployC2f
deploy/easydeploy/backbone/common.py:6
ClassEfficientCSPLayerWithTwoConv
Sigmoid-attention based CSP layer with two convolution layers.
yolo_world/models/layers/yolo_bricks.py:551
ClassHuggingCLIPLanguageBackbone
yolo_world/models/backbones/mm_backbone.py:59
ClassHuggingVisionBackbone
yolo_world/models/backbones/mm_backbone.py:15
ClassImagePoolingAttentionModule
yolo_world/models/layers/yolo_bricks.py:427
ClassLoadText
yolo_world/datasets/transformers/mm_transforms.py:101
ClassMaxSigmoidCSPLayerWithTwoConv
Sigmoid-attention based CSP layer with two convolution layers.
yolo_world/models/layers/yolo_bricks.py:257
ClassMultiModalDataset
Multi-modal dataset.
yolo_world/datasets/mm_dataset.py:14
ClassMultiModalMixedDataset
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
ClassMultiModalMosaic
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
ClassMultiModalMosaic9
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
ClassMultiModalYOLOBackbone
yolo_world/models/backbones/mm_backbone.py:192
ClassONNXNMSop
deploy/easydeploy/nms/ort_nms.py:100
ClassPseudoLanguageBackbone
Pseudo Language Backbone Args: text_embed_path (str): path to the text embedding file
yolo_world/models/backbones/mm_backbone.py:141
ClassRandomLoadText
yolo_world/datasets/transformers/mm_transforms.py:11
ClassRepConvMaxSigmoidCSPLayerWithTwoConv
Sigmoid-attention based CSP layer with two convolution layers.
yolo_world/models/layers/yolo_bricks.py:370
ClassRepMaxSigmoidCSPLayerWithTwoConv
Sigmoid-attention based CSP layer with two convolution layers.
yolo_world/models/layers/yolo_bricks.py:313
ClassRepYOLOWorldHeadModule
yolo_world/models/dense_heads/yolo_world_head.py:293
ClassSimpleYOLOWorldDetector
Implementation of YOLO World Series
yolo_world/models/detectors/yolo_world.py:109
ClassTASK_TYPE
deploy/easydeploy/examples/config.py:4
ClassTRTEfficientNMSop
deploy/easydeploy/nms/trt_nms.py:10
ClassTRTbatchedNMSop
TensorRT NMS operation.
deploy/easydeploy/nms/trt_nms.py:61
ClassV3DetDataset
Objects365 v1 dataset for detection.
yolo_world/datasets/yolov5_v3det.py:34
ClassYOLOWDetDataPreprocessor
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
ClassYOLOWorldDetector
Implementation of YOLOW Series
yolo_world/models/detectors/yolo_world.py:12
ClassYOLOWorldDualPAFPN
Path Aggregation Network used in YOLO World v8.
yolo_world/models/necks/yolo_world_pafpn.py:150
ClassYOLOWorldHead
YOLO-World Head
yolo_world/models/dense_heads/yolo_world_head.py:348
ClassYOLOWorldHeadModule
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
ClassYOLOWorldImageDetector
Implementation of YOLO World Series
yolo_world/models/detectors/yolo_world_image.py:164
ClassYOLOWorldPAFPN
Path Aggregation Network used in YOLO World Following YOLOv8 PAFPN, including text to image fusion
yolo_world/models/necks/yolo_world_pafpn.py:16
ClassYOLOWorldSegAssigner
yolo_world/models/assigner/task_aligned_assigner.py:9
ClassYOLOWorldSegHead
yolo_world/models/dense_heads/yolo_world_seg_head.py:216
ClassYOLOWorldSegHeadModule
yolo_world/models/dense_heads/yolo_world_seg_head.py:30
ClassYOLOWv5OptimizerConstructor
YOLO World v5 constructor for optimizers.
yolo_world/engine/optimizers/yolow_v5_optim_constructor.py:19
ClassYOLOXMultiModalMixUp
MixUp data augmentation for YOLOX. .. code:: text mixup transform +---------------+--------------+
yolo_world/datasets/transformers/mm_mix_img_transforms.py:941
ClassYOLOv5GeneralGroundingDataset
Mixed grounding dataset.
yolo_world/datasets/yolov5_cc3m_grounding.py:13
ClassYOLOv5LVISV1Dataset
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
ClassYOLOv5MixedGroundingDataset
Mixed grounding dataset.
yolo_world/datasets/yolov5_mixed_grounding.py:13
ClassYOLOv5MultiModalMixUp
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
ClassYOLOv5Objects365V1Dataset
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
ClassYOLOv5Objects365V2Dataset
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
ClassYOLOv5V3DetDataset
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