Python AttributeError:找不到字段'；pred#u面具'；在给定的情况下

我在colab上训练了一个带有视网膜神经网络Detectron2的模型。培训是成功的，但我无法在图像上测试它

代码：

错误：

/内容/驱动器/我的 驱动/检测/测试/wh_dresscode_4969_jpg.rf.5a2702518691d3d6f9cef7ad7977e5c8.jpg 数据类型： AttributeError回溯（最近的调用） 最后)

AttributeError:在给定实例中找不到字段“pred_masks”

---

似乎您的模型经过训练，只能预测边界框；也就是说，你不能像maskrcnn那样预测分段。如果是这种情况，您可以从

Visualizer

中删除

instance\u mode=ColorMode.IMAGE\u BW

from detectron2.utils.visualizer import ColorMode
import glob

for imageName in glob.glob('/content/drive/My Drive/Detectron/test/*jpg'):
    print(imageName)
    im = cv2.imread(imageName)
    outputs = predictor(im)
    v = Visualizer(im[:, :, ::-1],
                metadata=test_metadata, 
                scale=0.8,
                instance_mode=ColorMode.IMAGE_BW
                 )
    out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
    cv2_imshow(out.get_image()[:, :, ::-1])