助睿AI平台是否支持深度学习图像分类模型训练?

手头有几千张猫和狗的照片,想训练一个分类模型。助睿AI平台支持深度学习图像训练吗?

20 个赞

支持。

先使用「图像(Image)」控件加载数据集(支持 ImageNet/VOC/COCO 格式),连接「图像分类(Image Classification)」控件,选择 PyTorch 或 TensorFlow 计算框架,可选 ResNet50 等主流模型,配置迭代轮数、学习率、批大小后开始训练,再连接「训练日志」控件查看训练过程。

训练完成后还可通过「预测(Infer)」控件连接测试集进行推理验证。