bobo0810 / OpenMM

OpenMM最佳食用手册

OpenMM最佳食用手册

• 该仓库收录于PytorchNetHub
• 荣誉证书

MMEngine

• Hook用法 教程1 教程2

MMCV

• 验证图像完整性 教程

MMPretrain

• 下游识别模型加载自监督预训练权重 教程
• MM库训练时加载预训练权重 教程

MMYOLO目标检测

---

以下内容已过时

## MMClassification图像识别

整体框架图

配置文件-官方教程

优势

• 模型库：支持内置库、timm、huggingface
• 任务：支持单任务、多任务、TTA测试等

用法

• 启用timm模型库

  model = dict(
      _delete_=True,
      type="TimmClassifier",
      model_name="swinv2_base_window16_256",
      pretrained=True, # timm接口参数
      loss=xxx,
      train_cfg=xxx,
  )

• 命令行改参数

  bash ./tools/dist_train.sh  xx.py  --amp  --cfg-options train_dataloader.batch_size=12

MMSelfSup自监督

• 比较两个图像的相似度 代码
• 自定义数据集训练 代码