[英]AttributeError: Cannot find field 'pred_masks' in the given Instances
I trained a model with RetinaNet Detectron2 on colab.我在 colab 上用 RetinaNet Detectron2 训练了一个模型。 The training is successful but i am unable to test it on images.
培训成功,但我无法在图像上对其进行测试。
Code:代码:
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])
Error:错误:
/content/drive/My Drive/Detectron/test/wh_dresscode_4969_jpg.rf.5a2702518691d3d6f9cef7ad7977e5c8.jpg data type: <class 'detectron2.structures.instances.Instances'> AttributeError Traceback (most recent call last)
/content/drive/My Drive/Detectron/test/wh_dresscode_4969_jpg.rf.5a2702518691d3d6f9cef7ad7977e5c8.jpg 数据类型:<class 'detectron2.structures.instances.Instances'> AttributeError Traceback(最近一次调用)
AttributeError: Cannot find field 'pred_masks' in the given Instances!
AttributeError: 在给定的实例中找不到字段“pred_masks”!
it seems that your model was trained to predict only bounding boxes;似乎您的模型被训练为仅预测边界框; that is you do not predict segmentations masks as in maskrcnn.
也就是说,您不像在 maskrcnn 中那样预测分段掩码。 If that is the case, you can remove
instance_mode=ColorMode.IMAGE_BW
from Visualizer
如果是这种情况,您可以从
Visualizer
删除instance_mode=ColorMode.IMAGE_BW
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.