Mech-DLK 2.2.1 更新说明

新增功能

“图像分类”模块新增类激活图显示功能

模型训练完成后,点击 生成类激活图 可以通过热力图的形式体现特征的权重,模型基于这些特征将该图像分为此类别。颜色越红的区域,在分为此类别的过程中所占的权重越大。

CPU 版本模型的验证及导出功能

  • 图像分类、目标检测:训练完成后,导出模型前可以将部署设备选为 CPU 或 GPU。

  • 实例分割:训练模型前,设置训练参数。导出模型时,可以在 CPU/GPU 之间选择部署设备,具体如下:

    • CPU 轻量模型:训练模型前,将训练参数 模型类型 选为 轻量(推荐使用 CPU 部署) ,导出模型部署时,可以将 部署设备 选为 CPUGPU

    • GPU 标准模型:训练模型前,将训练参数 模型类型 选为 标准(推荐使用 GPU 部署) ,导出模型部署时,推荐将 部署设备 选为 GPU