LICENSE
MANIFEST.in
README.md
setup.cfg
setup.py
mmpretrain/__init__.py
mmpretrain/registry.py
mmpretrain/version.py
mmpretrain.egg-info/PKG-INFO
mmpretrain.egg-info/SOURCES.txt
mmpretrain.egg-info/dependency_links.txt
mmpretrain.egg-info/not-zip-safe
mmpretrain.egg-info/requires.txt
mmpretrain.egg-info/top_level.txt
mmpretrain/apis/__init__.py
mmpretrain/apis/base.py
mmpretrain/apis/feature_extractor.py
mmpretrain/apis/image_caption.py
mmpretrain/apis/image_classification.py
mmpretrain/apis/image_retrieval.py
mmpretrain/apis/model.py
mmpretrain/apis/multimodal_retrieval.py
mmpretrain/apis/nlvr.py
mmpretrain/apis/utils.py
mmpretrain/apis/visual_grounding.py
mmpretrain/apis/visual_question_answering.py
mmpretrain/datasets/__init__.py
mmpretrain/datasets/base_dataset.py
mmpretrain/datasets/builder.py
mmpretrain/datasets/caltech101.py
mmpretrain/datasets/categories.py
mmpretrain/datasets/cifar.py
mmpretrain/datasets/coco_caption.py
mmpretrain/datasets/coco_retrieval.py
mmpretrain/datasets/coco_vqa.py
mmpretrain/datasets/cub.py
mmpretrain/datasets/custom.py
mmpretrain/datasets/dataset_wrappers.py
mmpretrain/datasets/dtd.py
mmpretrain/datasets/fgvcaircraft.py
mmpretrain/datasets/flamingo.py
mmpretrain/datasets/flickr30k_caption.py
mmpretrain/datasets/flickr30k_retrieval.py
mmpretrain/datasets/flowers102.py
mmpretrain/datasets/food101.py
mmpretrain/datasets/gqa_dataset.py
mmpretrain/datasets/iconqa.py
mmpretrain/datasets/imagenet.py
mmpretrain/datasets/infographic_vqa.py
mmpretrain/datasets/inshop.py
mmpretrain/datasets/minigpt4_dataset.py
mmpretrain/datasets/mnist.py
mmpretrain/datasets/multi_label.py
mmpretrain/datasets/multi_task.py
mmpretrain/datasets/nlvr2.py
mmpretrain/datasets/nocaps.py
mmpretrain/datasets/ocr_vqa.py
mmpretrain/datasets/oxfordiiitpet.py
mmpretrain/datasets/places205.py
mmpretrain/datasets/refcoco.py
mmpretrain/datasets/scienceqa.py
mmpretrain/datasets/stanfordcars.py
mmpretrain/datasets/sun397.py
mmpretrain/datasets/textvqa.py
mmpretrain/datasets/utils.py
mmpretrain/datasets/vg_vqa.py
mmpretrain/datasets/visual_genome.py
mmpretrain/datasets/vizwiz.py
mmpretrain/datasets/voc.py
mmpretrain/datasets/vsr.py
mmpretrain/datasets/samplers/__init__.py
mmpretrain/datasets/samplers/repeat_aug.py
mmpretrain/datasets/samplers/sequential.py
mmpretrain/datasets/transforms/__init__.py
mmpretrain/datasets/transforms/auto_augment.py
mmpretrain/datasets/transforms/formatting.py
mmpretrain/datasets/transforms/processing.py
mmpretrain/datasets/transforms/utils.py
mmpretrain/datasets/transforms/wrappers.py
mmpretrain/engine/__init__.py
mmpretrain/engine/hooks/__init__.py
mmpretrain/engine/hooks/class_num_check_hook.py
mmpretrain/engine/hooks/densecl_hook.py
mmpretrain/engine/hooks/ema_hook.py
mmpretrain/engine/hooks/margin_head_hooks.py
mmpretrain/engine/hooks/precise_bn_hook.py
mmpretrain/engine/hooks/retriever_hooks.py
mmpretrain/engine/hooks/simsiam_hook.py
mmpretrain/engine/hooks/swav_hook.py
mmpretrain/engine/hooks/switch_recipe_hook.py
mmpretrain/engine/hooks/visualization_hook.py
mmpretrain/engine/hooks/warmup_param_hook.py
mmpretrain/engine/optimizers/__init__.py
mmpretrain/engine/optimizers/adan_t.py
mmpretrain/engine/optimizers/lamb.py
mmpretrain/engine/optimizers/lars.py
mmpretrain/engine/optimizers/layer_decay_optim_wrapper_constructor.py
mmpretrain/engine/runners/__init__.py
mmpretrain/engine/runners/retrieval_loop.py
mmpretrain/engine/schedulers/__init__.py
mmpretrain/engine/schedulers/weight_decay_scheduler.py
mmpretrain/evaluation/__init__.py
mmpretrain/evaluation/functional/__init__.py
mmpretrain/evaluation/metrics/ANLS.py
mmpretrain/evaluation/metrics/__init__.py
mmpretrain/evaluation/metrics/caption.py
mmpretrain/evaluation/metrics/gqa.py
mmpretrain/evaluation/metrics/multi_label.py
mmpretrain/evaluation/metrics/multi_task.py
mmpretrain/evaluation/metrics/nocaps.py
mmpretrain/evaluation/metrics/retrieval.py
mmpretrain/evaluation/metrics/scienceqa.py
mmpretrain/evaluation/metrics/shape_bias_label.py
mmpretrain/evaluation/metrics/single_label.py
mmpretrain/evaluation/metrics/visual_grounding_eval.py
mmpretrain/evaluation/metrics/voc_multi_label.py
mmpretrain/evaluation/metrics/vqa.py
mmpretrain/models/__init__.py
mmpretrain/models/builder.py
mmpretrain/models/backbones/__init__.py
mmpretrain/models/backbones/alexnet.py
mmpretrain/models/backbones/base_backbone.py
mmpretrain/models/backbones/beit.py
mmpretrain/models/backbones/conformer.py
mmpretrain/models/backbones/convmixer.py
mmpretrain/models/backbones/convnext.py
mmpretrain/models/backbones/cspnet.py
mmpretrain/models/backbones/davit.py
mmpretrain/models/backbones/deit.py
mmpretrain/models/backbones/deit3.py
mmpretrain/models/backbones/densenet.py
mmpretrain/models/backbones/edgenext.py
mmpretrain/models/backbones/efficientformer.py
mmpretrain/models/backbones/efficientnet.py
mmpretrain/models/backbones/efficientnet_v2.py
mmpretrain/models/backbones/hivit.py
mmpretrain/models/backbones/hornet.py
mmpretrain/models/backbones/hrnet.py
mmpretrain/models/backbones/inception_v3.py
mmpretrain/models/backbones/lenet.py
mmpretrain/models/backbones/levit.py
mmpretrain/models/backbones/mixmim.py
mmpretrain/models/backbones/mlp_mixer.py
mmpretrain/models/backbones/mobilenet_v2.py
mmpretrain/models/backbones/mobilenet_v3.py
mmpretrain/models/backbones/mobileone.py
mmpretrain/models/backbones/mobilevit.py
mmpretrain/models/backbones/mvit.py
mmpretrain/models/backbones/poolformer.py
mmpretrain/models/backbones/regnet.py
mmpretrain/models/backbones/replknet.py
mmpretrain/models/backbones/repmlp.py
mmpretrain/models/backbones/repvgg.py
mmpretrain/models/backbones/res2net.py
mmpretrain/models/backbones/resnest.py
mmpretrain/models/backbones/resnet.py
mmpretrain/models/backbones/resnet_cifar.py
mmpretrain/models/backbones/resnext.py
mmpretrain/models/backbones/revvit.py
mmpretrain/models/backbones/riformer.py
mmpretrain/models/backbones/seresnet.py
mmpretrain/models/backbones/seresnext.py
mmpretrain/models/backbones/shufflenet_v1.py
mmpretrain/models/backbones/shufflenet_v2.py
mmpretrain/models/backbones/sparse_convnext.py
mmpretrain/models/backbones/sparse_resnet.py
mmpretrain/models/backbones/swin_transformer.py
mmpretrain/models/backbones/swin_transformer_v2.py
mmpretrain/models/backbones/t2t_vit.py
mmpretrain/models/backbones/timm_backbone.py
mmpretrain/models/backbones/tinyvit.py
mmpretrain/models/backbones/tnt.py
mmpretrain/models/backbones/twins.py
mmpretrain/models/backbones/van.py
mmpretrain/models/backbones/vgg.py
mmpretrain/models/backbones/vig.py
mmpretrain/models/backbones/vision_transformer.py
mmpretrain/models/backbones/vit_eva02.py
mmpretrain/models/backbones/vit_sam.py
mmpretrain/models/backbones/xcit.py
mmpretrain/models/classifiers/__init__.py
mmpretrain/models/classifiers/base.py
mmpretrain/models/classifiers/hugging_face.py
mmpretrain/models/classifiers/image.py
mmpretrain/models/classifiers/timm.py
mmpretrain/models/heads/__init__.py
mmpretrain/models/heads/beitv1_head.py
mmpretrain/models/heads/beitv2_head.py
mmpretrain/models/heads/cae_head.py
mmpretrain/models/heads/cls_head.py
mmpretrain/models/heads/conformer_head.py
mmpretrain/models/heads/contrastive_head.py
mmpretrain/models/heads/deit_head.py
mmpretrain/models/heads/efficientformer_head.py
mmpretrain/models/heads/grounding_head.py
mmpretrain/models/heads/itc_head.py
mmpretrain/models/heads/itm_head.py
mmpretrain/models/heads/itpn_clip_head.py
mmpretrain/models/heads/latent_heads.py
mmpretrain/models/heads/levit_head.py
mmpretrain/models/heads/linear_head.py
mmpretrain/models/heads/mae_head.py
mmpretrain/models/heads/margin_head.py
mmpretrain/models/heads/mim_head.py
mmpretrain/models/heads/mixmim_head.py
mmpretrain/models/heads/mocov3_head.py
mmpretrain/models/heads/multi_label_cls_head.py
mmpretrain/models/heads/multi_label_csra_head.py
mmpretrain/models/heads/multi_label_linear_head.py
mmpretrain/models/heads/multi_task_head.py
mmpretrain/models/heads/seq_gen_head.py
mmpretrain/models/heads/simmim_head.py
mmpretrain/models/heads/spark_head.py
mmpretrain/models/heads/stacked_head.py
mmpretrain/models/heads/swav_head.py
mmpretrain/models/heads/vig_head.py
mmpretrain/models/heads/vision_transformer_head.py
mmpretrain/models/heads/vqa_head.py
mmpretrain/models/losses/__init__.py
mmpretrain/models/losses/asymmetric_loss.py
mmpretrain/models/losses/cae_loss.py
mmpretrain/models/losses/cosine_similarity_loss.py
mmpretrain/models/losses/cross_correlation_loss.py
mmpretrain/models/losses/cross_entropy_loss.py
mmpretrain/models/losses/focal_loss.py
mmpretrain/models/losses/label_smooth_loss.py
mmpretrain/models/losses/reconstruction_loss.py
mmpretrain/models/losses/seesaw_loss.py
mmpretrain/models/losses/swav_loss.py
mmpretrain/models/losses/utils.py
mmpretrain/models/multimodal/__init__.py
mmpretrain/models/multimodal/blip/__init__.py
mmpretrain/models/multimodal/blip/blip_caption.py
mmpretrain/models/multimodal/blip/blip_grounding.py
mmpretrain/models/multimodal/blip/blip_nlvr.py
mmpretrain/models/multimodal/blip/blip_retrieval.py
mmpretrain/models/multimodal/blip/blip_vqa.py
mmpretrain/models/multimodal/blip/language_model.py
mmpretrain/models/multimodal/blip2/Qformer.py
mmpretrain/models/multimodal/blip2/__init__.py
mmpretrain/models/multimodal/blip2/blip2_caption.py
mmpretrain/models/multimodal/blip2/blip2_opt_vqa.py
mmpretrain/models/multimodal/blip2/blip2_retriever.py
mmpretrain/models/multimodal/blip2/modeling_opt.py
mmpretrain/models/multimodal/chinese_clip/__init__.py
mmpretrain/models/multimodal/chinese_clip/bert.py
mmpretrain/models/multimodal/chinese_clip/chinese_clip.py
mmpretrain/models/multimodal/chinese_clip/utils.py
mmpretrain/models/multimodal/clip/__init__.py
mmpretrain/models/multimodal/clip/clip.py
mmpretrain/models/multimodal/clip/clip_transformer.py
mmpretrain/models/multimodal/clip/utils.py
mmpretrain/models/multimodal/flamingo/__init__.py
mmpretrain/models/multimodal/flamingo/adapter.py
mmpretrain/models/multimodal/flamingo/flamingo.py
mmpretrain/models/multimodal/flamingo/modules.py
mmpretrain/models/multimodal/flamingo/utils.py
mmpretrain/models/multimodal/llava/__init__.py
mmpretrain/models/multimodal/llava/llava.py
mmpretrain/models/multimodal/llava/modules.py
mmpretrain/models/multimodal/minigpt4/__init__.py
mmpretrain/models/multimodal/minigpt4/minigpt4.py
mmpretrain/models/multimodal/ofa/__init__.py
mmpretrain/models/multimodal/ofa/ofa.py
mmpretrain/models/multimodal/ofa/ofa_modules.py
mmpretrain/models/multimodal/otter/__init__.py
mmpretrain/models/multimodal/otter/otter.py
mmpretrain/models/multimodal/ram/__init__.py
mmpretrain/models/multimodal/ram/bert.py
mmpretrain/models/multimodal/ram/gradio_demo.py
mmpretrain/models/multimodal/ram/openset_utils.py
mmpretrain/models/multimodal/ram/ram.py
mmpretrain/models/multimodal/ram/utils.py
mmpretrain/models/multimodal/ram/config/__init__.py
mmpretrain/models/multimodal/ram/config/ram_swin_large_14m.py
mmpretrain/models/multimodal/ram/run/__init__.py
mmpretrain/models/multimodal/ram/run/inference.py
mmpretrain/models/necks/__init__.py
mmpretrain/models/necks/beitv2_neck.py
mmpretrain/models/necks/cae_neck.py
mmpretrain/models/necks/densecl_neck.py
mmpretrain/models/necks/gap.py
mmpretrain/models/necks/gem.py
mmpretrain/models/necks/hr_fuse.py
mmpretrain/models/necks/itpn_neck.py
mmpretrain/models/necks/linear_neck.py
mmpretrain/models/necks/mae_neck.py
mmpretrain/models/necks/milan_neck.py
mmpretrain/models/necks/mixmim_neck.py
mmpretrain/models/necks/mocov2_neck.py
mmpretrain/models/necks/nonlinear_neck.py
mmpretrain/models/necks/simmim_neck.py
mmpretrain/models/necks/spark_neck.py
mmpretrain/models/necks/swav_neck.py
mmpretrain/models/peft/__init__.py
mmpretrain/models/peft/lora.py
mmpretrain/models/retrievers/__init__.py
mmpretrain/models/retrievers/base.py
mmpretrain/models/retrievers/image2image.py
mmpretrain/models/selfsup/__init__.py
mmpretrain/models/selfsup/barlowtwins.py
mmpretrain/models/selfsup/base.py
mmpretrain/models/selfsup/beit.py
mmpretrain/models/selfsup/byol.py
mmpretrain/models/selfsup/cae.py
mmpretrain/models/selfsup/densecl.py
mmpretrain/models/selfsup/eva.py
mmpretrain/models/selfsup/itpn.py
mmpretrain/models/selfsup/mae.py
mmpretrain/models/selfsup/maskfeat.py
mmpretrain/models/selfsup/mff.py
mmpretrain/models/selfsup/milan.py
mmpretrain/models/selfsup/mixmim.py
mmpretrain/models/selfsup/moco.py
mmpretrain/models/selfsup/mocov3.py
mmpretrain/models/selfsup/simclr.py
mmpretrain/models/selfsup/simmim.py
mmpretrain/models/selfsup/simsiam.py
mmpretrain/models/selfsup/spark.py
mmpretrain/models/selfsup/swav.py
mmpretrain/models/tta/__init__.py
mmpretrain/models/tta/score_tta.py
mmpretrain/models/utils/__init__.py
mmpretrain/models/utils/attention.py
mmpretrain/models/utils/batch_shuffle.py
mmpretrain/models/utils/box_utils.py
mmpretrain/models/utils/channel_shuffle.py
mmpretrain/models/utils/clip_generator_helper.py
mmpretrain/models/utils/data_preprocessor.py
mmpretrain/models/utils/ema.py
mmpretrain/models/utils/embed.py
mmpretrain/models/utils/helpers.py
mmpretrain/models/utils/huggingface.py
mmpretrain/models/utils/inverted_residual.py
mmpretrain/models/utils/layer_scale.py
mmpretrain/models/utils/make_divisible.py
mmpretrain/models/utils/norm.py
mmpretrain/models/utils/position_encoding.py
mmpretrain/models/utils/res_layer_extra_norm.py
mmpretrain/models/utils/se_layer.py
mmpretrain/models/utils/sparse_modules.py
mmpretrain/models/utils/swiglu_ffn.py
mmpretrain/models/utils/tokenizer.py
mmpretrain/models/utils/vector_quantizer.py
mmpretrain/models/utils/batch_augments/__init__.py
mmpretrain/models/utils/batch_augments/cutmix.py
mmpretrain/models/utils/batch_augments/mixup.py
mmpretrain/models/utils/batch_augments/resizemix.py
mmpretrain/models/utils/batch_augments/wrapper.py
mmpretrain/structures/__init__.py
mmpretrain/structures/data_sample.py
mmpretrain/structures/multi_task_data_sample.py
mmpretrain/structures/utils.py
mmpretrain/utils/__init__.py
mmpretrain/utils/analyze.py
mmpretrain/utils/collect_env.py
mmpretrain/utils/dependency.py
mmpretrain/utils/misc.py
mmpretrain/utils/progress.py
mmpretrain/utils/setup_env.py
mmpretrain/visualization/__init__.py
mmpretrain/visualization/utils.py
mmpretrain/visualization/visualizer.py
requirements/docs.txt
requirements/mminstall.txt
requirements/multimodal.txt
requirements/optional.txt
requirements/readthedocs.txt
requirements/runtime.txt
requirements/tests.txt
tests/test_tools.py