一种厨余垃圾遮挡物体识别和遮挡关系判断方法

本发明公开了一种厨余垃圾遮挡物体识别和遮挡关系判断方法，构建厨余垃圾图像的遮挡数据集，标有其中各目标物体的可见掩码、完整掩码、类别；首先，利用遮挡数据集训练实例分割模型和掩码还原模型；实例分割模型的输入为厨余垃圾图像，输出为各目标物体的类别和可见掩码；掩码还原模型的输入为实例分割模型输出的可见掩码和厨余垃圾图像，输出为完整掩码；其次，将待识别判断的厨余垃圾图像，使用训练好的实例分割模型和掩码还原模型，获得各目标物体的类别、可见掩码、完整掩码；最后，结合可见掩码和完整掩码判断各目标物体是否被遮挡及遮挡关系。本发明对可回收垃圾进行实例分割和掩码还原，能获得物体类别、位置、遮挡关系和完整的形状信息。