Mech-DLK V2.2.1 Release Notes

New Features

Added the Function of Showing the Class Activation Maps for Module “Classification”

After the model is trained, click Generate CAM. The class activation maps show the weights of the features in the form of heat maps; the model classifies an image into its class according to these features. Image regions with warmer colors have higher weights for classifying the image into its class.

Verification and Export of CPU models

  • Classification, Object Detection: After training is complete, select the deployment device as CPU or GPU before exporting the model.

  • Instance segmentation: Before training the model, set the training parameters. When exporting a model, select the deployment device as CPU/GPU:

    • CPU lightweight model : Before training the model, set the training parameter Model type to Lite (better with CPU deployment). When exporting the model for deployment, set Deployment device to CPU or GPU.

    • GPU standard model : Before training the model, set the training parameter Model type to Normal (better with GPU deployment). When exporting the model for deployment, it is recommended to set Deployment device to GPU.