mmpretrain 增加回归任务组件
mmpretrain 原生只服务分类任务。本文把它魔改出完整的回归能力:自定义数据集、预处理器、回归器、回归头、MAE 损失与评估指标、推理器,一路改到 gradio 演示,全程沿 mm 系列的注册器机制进行,不破坏框架原有结构。
openmmlab介绍
openmmlab系列框架在2023年全面进入2.0时代,官方把自监督mmselfsup和分类mmclassification两个项目合并成mmpretrain。
openmmlab 2.0系列底层是有一些变化的,最大变化是从mmcv中抽离出功能更强大的mmengine。新的项目我还没仔细看过,目...