基于细粒度图像分类的排水管道缺陷分类分级方法及系统

本申请公开了一种基于细粒度图像分类的排水管道缺陷分类分级方法及系统，所述方法包括：构建并训练细粒度缺陷分级分类模型；利用训练好的模型对输入的排水管道图片进行缺陷分类分级识别；通过图像特征提取主干网络对输入的排水管道图片进行特征提取，将提取的特征同时输入至缺陷分类头和N个特定缺陷类别分级头；通过缺陷分类头对输入的特征进行缺陷分类，以N个输出结果取最大值对应的类别为识别的缺陷类别；通过各特定缺陷类别分级头对输入的特征进行对应类别的缺陷分级，获取识别的缺陷类别对应的分级头的输出，计算缺陷等级属于第k级别的概率，以概率最大的级别为排水管道图片缺陷级别。本申请能够自动、准确地进行排水管道缺陷分类分级。