图像分类工具包MMClassification
具有丰富的模型,包括卷积神经网络,如VGG;轻量化卷积网络,如MobileNet V2/V3;Transformer模型,如ViT等。
在数据集方面,支持MNIST/CIFAR,以及自定义数据集。
在训练技巧与策略方面,具有优化器与学习策略,数据增强策略。
也包含易用的工具,如大量预制配置文件、预训练模型,Python推理API,模型训练工具,CAM可解释性分析,部署工具。
图像分类工具包MMClassification
具有丰富的模型,包括卷积神经网络,如VGG;轻量化卷积网络,如MobileNet V2/V3;Transformer模型,如ViT等。
在数据集方面,支持MNIST/CIFAR,以及自定义数据集。
在训练技巧与策略方面,具有优化器与学习策略,数据增强策略。
也包含易用的工具,如大量预制配置文件、预训练模型,Python推理API,模型训练工具,CAM可解释性分析,部署工具。