一种图像分类模型构建方法及装置

本申请提供一种图像分类模型构建方法及装置，训练数据集可以包括多个训练图像，训练图像具有训练标签，基于训练数据集可以构建图像分类模型的各个分类节点，分类节点具有分流规则。在构建各个分类节点的过程中，若各个分类节点中的目标节点对应的缺陷图像对应多个训练标签，则构建目标节点的多个子节点，即图像分类模型为树状结构，分类节点的数量根据训练过程确定，具有较低的参数量，因此具有更快的分类速度。目标节点用于根据目标节点的分流规则，对目标节点对应的缺陷图像进行分类，得到与多个子节点一一对应的多组缺陷图像，这样可以通过各个分类节点依次对缺陷图像进行分类，从而准确快速确定缺陷图像的缺陷类别，提高分类精度和速度。