MM 系列开源库注册表

（注意：本文档是通过 .dev_scripts/print_registers.py 脚本自动生成）

MMdetection (3.0.0rc6)

MMdetection Module Components

visualizer	optimizer constructor	loop	parameter scheduler	data sampler
DetLocalVisualizer	LearningRateDecayOptimizerConstructor	TeacherStudentValLoop	QuadraticWarmupParamScheduler QuadraticWarmupLR QuadraticWarmupMomentum	AspectRatioBatchSampler ClassAwareSampler MultiSourceSampler GroupMultiSourceSampler

metric	hook	dataset	task util (part 1)	task util (part 2)
CityScapesMetric CocoMetric CocoOccludedSeparatedMetric CocoPanopticMetric CrowdHumanMetric DumpDetResults DumpProposals LVISMetric OpenImagesMetric VOCMetric	CheckInvalidLossHook MeanTeacherHook MemoryProfilerHook NumClassCheckHook PipelineSwitchHook SetEpochInfoHook SyncNormHook DetVisualizationHook YOLOXModeSwitchHook FastStopTrainingHook	BaseDetDataset CocoDataset CityscapesDataset CocoPanopticDataset CrowdHumanDataset MultiImageMixDataset DeepFashionDataset LVISV05Dataset LVISDataset LVISV1Dataset Objects365V1Dataset Objects365V2Dataset OpenImagesDataset OpenImagesChallengeDataset XMLDataset VOCDataset WIDERFaceDataset	MaxIoUAssigner ApproxMaxIoUAssigner ATSSAssigner CenterRegionAssigner DynamicSoftLabelAssigner GridAssigner HungarianAssigner BboxOverlaps2D BBoxL1Cost IoUCost ClassificationCost FocalLossCost DiceCost CrossEntropyLossCost MultiInstanceAssigner	PointAssigner AnchorGenerator SSDAnchorGenerator LegacyAnchorGenerator LegacySSDAnchorGenerator YOLOAnchorGenerator PointGenerator MlvlPointGenerator RegionAssigner SimOTAAssigner TaskAlignedAssigner UniformAssigner BucketingBBoxCoder DeltaXYWHBBoxCoder

task util (part 3)	transform (part 1)	transform (part 2)	transform (part 3)
DistancePointBBoxCoder LegacyDeltaXYWHBBoxCoder PseudoBBoxCoder TBLRBBoxCoder YOLOBBoxCoder CombinedSampler RandomSampler InstanceBalancedPosSampler IoUBalancedNegSampler MaskPseudoSampler MultiInsRandomSampler OHEMSampler PseudoSampler ScoreHLRSampler	AutoAugment RandAugment ColorTransform Color Brightness Contrast Sharpness Solarize SolarizeAdd Posterize Equalize AutoContrast Invert PackDetInputs ToTensor ImageToTensor Transpose WrapFieldsToLists GeomTransform ShearX	ShearY Rotate TranslateX TranslateY InstaBoost LoadImageFromNDArray LoadMultiChannelImageFromFiles LoadAnnotations LoadPanopticAnnotations LoadProposals FilterAnnotations LoadEmptyAnnotations InferencerLoader Resize FixShapeResize RandomFlip RandomShift Pad RandomCrop	SegRescale PhotoMetricDistortion Expand MinIoURandomCrop Corrupt Albu RandomCenterCropPad CutOut Mosaic MixUp RandomAffine YOLOXHSVRandomAug CopyPaste RandomErasing CachedMosaic CachedMixUp MultiBranch RandomOrder ProposalBroadcaster

model (part 1)	model (part 2)	model (part 3)	model (part 4)
SiLU DropBlock ExpMomentumEMA SinePositionalEncoding LearnedPositionalEncoding DynamicConv MSDeformAttnPixelDecoder Linear NormedLinear NormedConv2d PixelDecoder TransformerEncoderPixelDecoder CSPDarknet CSPNeXt Darknet ResNet ResNetV1d DetectoRS_ResNet DetectoRS_ResNeXt EfficientNet	HourglassNet HRNet MobileNetV2 PyramidVisionTransformer PyramidVisionTransformerV2 ResNeXt RegNet Res2Net ResNeSt BFP ChannelMapper CSPNeXtPAFPN CTResNetNeck DilatedEncoder DyHead FPG FPN FPN_CARAFE HRFPN NASFPN	NASFCOS_FPN PAFPN RFP SSDNeck SSH YOLOV3Neck YOLOXPAFPN SSDVGG SwinTransformer TridentResNet DetDataPreprocessor BatchSyncRandomResize BatchFixedSizePad MultiBranchDataPreprocessor BatchResize BoxInstDataPreprocessor AnchorFreeHead AnchorHead ATSSHead FCOSHead	AutoAssignHead CondInstBboxHead CondInstMaskHead BoxInstBboxHead BoxInstMaskHead RPNHead StageCascadeRPNHead CascadeRPNHead CenterNetHead CenterNetUpdateHead CornerHead CentripetalHead DETRHead ConditionalDETRHead DABDETRHead DDODHead DeformableDETRHead DINOHead EmbeddingRPNHead FoveaHead

model (part 5)	model (part 6)	model (part 7)	model (part 8)
RetinaHead FreeAnchorRetinaHead AssociativeEmbeddingLoss BalancedL1Loss CrossEntropyLoss DiceLoss FocalLoss GaussianFocalLoss QualityFocalLoss DistributionFocalLoss GHMC GHMR IoULoss BoundedIoULoss GIoULoss DIoULoss CIoULoss EIoULoss KnowledgeDistillationKLDivLoss MSELoss	SeesawLoss SmoothL1Loss L1Loss VarifocalLoss FSAFHead GuidedAnchorHead GARetinaHead GARPNHead GFLHead PAAHead LADHead LDHead MaskFormerHead Mask2FormerHead NASFCOSHead PISARetinaHead SSDHead PISASSDHead RepPointsHead RetinaSepBNHead	RTMDetHead RTMDetSepBNHead RTMDetInsHead RTMDetInsSepBNHead SABLRetinaHead SOLOHead DecoupledSOLOHead DecoupledSOLOLightHead SOLOV2Head TOODHead VFNetHead YOLACTHead YOLACTProtonet YOLOV3Head YOLOFHead YOLOXHead SingleStageDetector ATSS AutoAssign	DetectionTransformer SingleStageInstanceSegmentor BoxInst TwoStageDetector CascadeRCNN CenterNet CondInst DETR ConditionalDETR CornerNet CrowdDet Detectron2Wrapper DABDETR DDOD DeformableDETR DINO FastRCNN FasterRCNN FCOS

model (part 9)	model (part 10)	model (part 11)	model (part 12)
FOVEA FSAF GFL GridRCNN HybridTaskCascade KnowledgeDistillationSingleStageDetector LAD MaskFormer Mask2Former MaskRCNN MaskScoringRCNN NASFCOS PAA TwoStagePanopticSegmentor PanopticFPN PointRend SparseRCNN QueryInst RepPointsDetector	RetinaNet RPN RTMDet SCNet SemiBaseDetector SoftTeacher SOLO SOLOv2 TOOD TridentFasterRCNN VFNet YOLACT YOLOV3 YOLOF YOLOX BBoxHead ConvFCBBoxHead Shared2FCBBoxHead Shared4Conv1FCBBoxHead	DIIHead DoubleConvFCBBoxHead MultiInstanceBBoxHead SABLHead SCNetBBoxHead CascadeRoIHead StandardRoIHead DoubleHeadRoIHead DynamicRoIHead GridRoIHead HybridTaskCascadeRoIHead FCNMaskHead CoarseMaskHead DynamicMaskHead FeatureRelayHead FusedSemanticHead GlobalContextHead GridHead HTCMaskHead	MaskPointHead MaskIoUHead SCNetMaskHead SCNetSemanticHead MaskScoringRoIHead MultiInstanceRoIHead PISARoIHead PointRendRoIHead GenericRoIExtractor SingleRoIExtractor SCNetRoIHead ResLayer SparseRoIHead TridentRoIHead BaseSemanticHead PanopticFPNHead BasePanopticFusionHead HeuristicFusionHead MaskFormerFusionHead

MMdetection Tools

tools/dataset_converters	tools/deployment	tools	tools/misc
pascal_voc.py images2coco.py cityscapes.py	mmdet2torchserve.py test_torchserver.py mmdet_handler.py	dist_test.sh slurm_test.sh test.py dist_train.sh train.py slurm_train.sh	download_dataset.py get_image_metas.py gen_coco_panoptic_test_info.py split_coco.py get_crowdhuman_id_hw.py print_config.py

tools/model_converters	tools/analysis_tools	.dev_scripts (part 1)	.dev_scripts (part 2)
upgrade_model_version.py upgrade_ssd_version.py detectron2_to_mmdet.py selfsup2mmdet.py detectron2pytorch.py regnet2mmdet.py publish_model.py	benchmark.py eval_metric.py robustness_eval.py confusion_matrix.py optimize_anchors.py browse_dataset.py test_robustness.py coco_error_analysis.py coco_occluded_separated_recall.py analyze_results.py analyze_logs.py get_flops.py	convert_test_benchmark_script.py gather_test_benchmark_metric.py benchmark_valid_flops.py benchmark_train.py test_benchmark.sh download_checkpoints.py benchmark_test_image.py covignore.cfg benchmark_full_models.txt test_init_backbone.py batch_train_list.txt diff_coverage_test.sh	batch_test_list.py linter.sh gather_train_benchmark_metric.py train_benchmark.sh benchmark_inference_fps.py benchmark_options.py check_links.py benchmark_test.py benchmark_train_models.txt convert_train_benchmark_script.py gather_models.py benchmark_filter.py

MMclassification (1.0.0rc5)

MMclassification Module Components

visualizer	data sampler	optimizer	batch augment	metric
ClsVisualizer	RepeatAugSampler	Adan Lamb	Mixup CutMix ResizeMix	Accuracy SingleLabelMetric MultiLabelMetric AveragePrecision MultiTasksMetric VOCMultiLabelMetric VOCAveragePrecision

hook	dataset	transform (part 1)	transform (part 2)
ClassNumCheckHook EMAHook SetAdaptiveMarginsHook PreciseBNHook PrepareProtoBeforeValLoopHook SwitchRecipeHook VisualizationHook	BaseDataset CIFAR10 CIFAR100 CUB CustomDataset KFoldDataset ImageNet ImageNet21k MNIST FashionMNIST MultiLabelDataset MultiTaskDataset VOC	AutoAugment RandAugment Shear Translate Rotate AutoContrast Invert Equalize Solarize SolarizeAdd Posterize Contrast ColorTransform Brightness Sharpness Cutout	PackClsInputs PackMultiTaskInputs Transpose ToPIL ToNumpy Collect RandomCrop RandomResizedCrop EfficientNetRandomCrop RandomErasing EfficientNetCenterCrop ResizeEdge ColorJitter Lighting Albumentations Albu

model (part 1)	model (part 2)	model (part 3)	model (part 4)
AlexNet ShiftWindowMSA ClsDataPreprocessor VisionTransformer BEiT Conformer ConvMixer ResNet ResNetV1c ResNetV1d ResNeXt CSPDarkNet CSPResNet CSPResNeXt DaViT DistilledVisionTransformer DeiT3 DenseNet PoolFormer EfficientFormer	EfficientNet EfficientNetV2 HorNet HRNet InceptionV3 LeNet5 MixMIMTransformer MlpMixer MobileNetV2 MobileNetV3 MobileOne MViT RegNet RepLKNet RepMLPNet RepVGG Res2Net ResNeSt ResNet_CIFAR RevVisionTransformer	SEResNet SEResNeXt ShuffleNetV1 ShuffleNetV2 SwinTransformer SwinTransformerV2 T2T_ViT TIMMBackbone TNT PCPVT SVT VAN VGG HuggingFaceClassifier ImageClassifier TimmClassifier ClsHead ConformerHead VisionTransformerClsHead DeiTClsHead	EfficientFormerClsHead LinearClsHead AsymmetricLoss CrossEntropyLoss FocalLoss LabelSmoothLoss SeesawLoss ArcFaceClsHead MultiLabelClsHead CSRAClsHead MultiLabelLinearClsHead MultiTaskHead StackedLinearClsHead GlobalAveragePooling GeneralizedMeanPooling HRFuseScales LinearReduction ImageToImageRetriever AverageClsScoreTTA

MMclassification Tools

tools/misc	tools/visualizations	tools/torchserve	.dev_scripts	tools/analysis_tools
verify_dataset.py print_config.py	browse_dataset.py vis_scheduler.py vis_cam.py	mmcls_handler.py mmcls2torchserve.py test_torchserver.py	compare_init.py ckpt_tree.py generate_readme.py	eval_metric.py analyze_results.py analyze_logs.py get_flops.py

.dev_scripts/benchmark_regression	tools	tools/model_converters (part 1)	tools/model_converters (part 2)
bench_train.yml 4-benchmark_speed.py 3-benchmark_train.py 1-benchmark_valid.py 2-benchmark_test.py	dist_test.sh slurm_test.sh test.py dist_train.sh train.py slurm_train.sh kfold-cross-valid.py	efficientnet_to_mmcls.py repvgg_to_mmcls.py clip_to_mmcls.py reparameterize_model.py shufflenetv2_to_mmcls.py van2mmcls.py hornet2mmcls.py mixmimx_to_mmcls.py edgenext_to_mmcls.py torchvision_to_mmcls.py twins2mmcls.py revvit_to_mmcls.py	convnext_to_mmcls.py replknet_to_mmcls.py efficientnetv2_to_mmcls.py mobilenetv2_to_mmcls.py mlpmixer_to_mmcls.py davit_to_mmcls.py vgg_to_mmcls.py deit3_to_mmcls.py eva_to_mmcls.py publish_model.py tinyvit_to_mmcls.py

MMsegmentation (1.0.0rc5)

MMsegmentation Module Components

task util	visualizer	hook	optimizer wrapper constructor	metric
OHEMPixelSampler	SegLocalVisualizer	SegVisualizationHook	LearningRateDecayOptimizerConstructor LayerDecayOptimizerConstructor	CitysMetric IoUMetric

dataset (part 1)	dataset (part 2)	transform (part 1)	transform (part 2)
BaseSegDataset ADE20KDataset ChaseDB1Dataset CityscapesDataset COCOStuffDataset DarkZurichDataset MultiImageMixDataset DecathlonDataset DRIVEDataset HRFDataset iSAIDDataset	ISPRSDataset LIPDataset LoveDADataset NightDrivingDataset PascalContextDataset PascalContextDataset59 PotsdamDataset STAREDataset SynapseDataset PascalVOCDataset	PackSegInputs LoadAnnotations LoadImageFromNDArray LoadBiomedicalImageFromFile LoadBiomedicalAnnotation LoadBiomedicalData ResizeToMultiple Rerange CLAHE RandomCrop RandomRotate RGB2Gray AdjustGamma	SegRescale PhotoMetricDistortion RandomCutOut RandomRotFlip RandomMosaic GenerateEdge ResizeShortestEdge BioMedical3DRandomCrop BioMedicalGaussianNoise BioMedicalGaussianBlur BioMedicalRandomGamma BioMedical3DPad BioMedical3DRandomFlip

model (part 1)	model (part 2)	model (part 3)	model (part 4)
VisionTransformer BEiT BiSeNetV1 BiSeNetV2 CGNet ERFNet CrossEntropyLoss DiceLoss FocalLoss LovaszLoss TverskyLoss ANNHead APCHead ASPPHead FCNHead CCHead DAHead DMHead DNLHead	DPTHead EMAHead EncHead FPNHead GCHead ISAHead KernelUpdator KernelUpdateHead IterativeDecodeHead LRASPPHead Mask2FormerHead MaskFormerHead NLHead OCRHead PointHead PSAHead PSPHead SegformerHead SegmenterMaskTransformerHead	DepthwiseSeparableASPPHead DepthwiseSeparableFCNHead SETRMLAHead SETRUPHead STDCHead UPerHead FastSCNN ResNet ResNetV1c ResNetV1d HRNet ICNet MAE MixVisionTransformer MobileNetV2 MobileNetV3 ResNeSt ResNeXt STDCNet	STDCContextPathNet SwinTransformer TIMMBackbone PCPVT SVT DeconvModule InterpConv UNet SegDataPreProcessor Feature2Pyramid FPN ICNeck JPU MLANeck MultiLevelNeck EncoderDecoder CascadeEncoderDecoder SegTTAModel

MMsegmentation Tools

tools/deployment	tools/misc	tools/torchserve	tools/analysis_tools
pytorch2torchscript.py	browse_dataset.py publish_model.py print_config.py	mmseg_handler.py mmseg2torchserve.py test_torchserve.py	benchmark.py confusion_matrix.py analyze_logs.py get_flops.py

tools	tools/model_converters	tools/dataset_converters
dist_test.sh slurm_test.sh test.py dist_train.sh train.py slurm_train.sh	swin2mmseg.py vitjax2mmseg.py twins2mmseg.py stdc2mmseg.py vit2mmseg.py mit2mmseg.py beit2mmseg.py	voc_aug.py hrf.py drive.py pascal_context.py vaihingen.py stare.py synapse.py isaid.py cityscapes.py loveda.py potsdam.py chase_db1.py coco_stuff164k.py coco_stuff10k.py

MMengine (0.6.0)

MMengine Module Components

log_processor	visualizer	metric	evaluator	runner
LogProcessor	Visualizer	DumpResults	Evaluator	Runner

optimizer wrapper constructor	Collate Functions	data sampler	vis_backend	dataset
DefaultOptimWrapperConstructor	pseudo_collate default_collate	DefaultSampler InfiniteSampler	LocalVisBackend WandbVisBackend TensorboardVisBackend	ConcatDataset RepeatDataset ClassBalancedDataset

optim_wrapper	loop	model_wrapper	model	weight initializer
OptimWrapper AmpOptimWrapper ApexOptimWrapper	EpochBasedTrainLoop IterBasedTrainLoop ValLoop TestLoop	DistributedDataParallel DataParallel MMDistributedDataParallel MMSeparateDistributedDataParallel	StochasticWeightAverage ExponentialMovingAverage MomentumAnnealingEMA BaseDataPreprocessor ImgDataPreprocessor BaseTTAModel ToyModel	Constant Xavier Normal TruncNormal Uniform Kaiming Caffe2Xavier Pretrained

hook	optimizer	parameter scheduler (part 1)	parameter scheduler (part 2)
CheckpointHook EMAHook EmptyCacheHook IterTimerHook LoggerHook NaiveVisualizationHook ParamSchedulerHook ProfilerHook NPUProfilerHook RuntimeInfoHook DistSamplerSeedHook SyncBuffersHook PrepareTTAHook	ASGD Adadelta Adagrad Adam AdamW Adamax LBFGS Optimizer RMSprop Rprop SGD SparseAdam ZeroRedundancyOptimizer	StepParamScheduler MultiStepParamScheduler ConstantParamScheduler ExponentialParamScheduler CosineAnnealingParamScheduler LinearParamScheduler PolyParamScheduler OneCycleParamScheduler CosineRestartParamScheduler ReduceOnPlateauParamScheduler ConstantLR CosineAnnealingLR ExponentialLR LinearLR MultiStepLR	StepLR PolyLR OneCycleLR CosineRestartLR ReduceOnPlateauLR ConstantMomentum CosineAnnealingMomentum ExponentialMomentum LinearMomentum MultiStepMomentum StepMomentum PolyMomentum CosineRestartMomentum ReduceOnPlateauMomentum

MMCV (2.0.0rc4)

MMCV Module Components

transform	model (part 1)	model (part 2)	model (part 3)
LoadImageFromFile LoadAnnotations Compose KeyMapper TransformBroadcaster RandomChoice RandomApply Normalize Resize Pad CenterCrop RandomGrayscale MultiScaleFlipAug TestTimeAug RandomChoiceResize RandomFlip RandomResize ToTensor ImageToTensor	ReLU LeakyReLU PReLU RReLU ReLU6 ELU Sigmoid Tanh SiLU Clamp Clip GELU ContextBlock Conv1d Conv2d Conv3d Conv Conv2dAdaptivePadding	BN BN1d BN2d BN3d SyncBN GN LN IN IN1d IN2d IN3d zero reflect replicate ConvModule ConvWS ConvAWS DropPath	Dropout GeneralizedAttention HSigmoid HSwish NonLocal2d Swish nearest bilinear pixel_shuffle deconv ConvTranspose2d deconv3d ConvTranspose3d MultiheadAttention FFN BaseTransformerLayer TransformerLayerSequence

MMCV Tools

.dev_scripts
check_installation.py