【发布时间】:2020-01-11 11:07:36
【问题描述】:
我正在开展一个项目,该项目使用 Mask-RCNN 在 SpaceNet 数据集上检测建筑物。当我运行这段代码时:
model.train(dataset_train, dataset_val,
learning_rate=config.LEARNING_RATE,
epochs=1,
layers='heads')
我收到了这个错误:
1772 batch_images[b] = mold_image(image.astype(np.float32), config)
-> 1773 batch_gt_class_ids[b, :gt_class_ids.shape[0]] = gt_class_ids
1774 batch_gt_boxes[b, :gt_boxes.shape[0]] = gt_boxes
1775 batch_gt_masks[b, :, :, :gt_masks.shape[-1]] = gt_masks
ValueError: 无法将输入数组从形状 (70) 广播到形状 (1)
当我调试变量:“gt_class_ids”和“batch_gt_class_ids”时, 我得到了这个:
【问题讨论】:
-
嗨,我在第 1775 行“batch_gt_masks[b, :, :, :gt_masks.shape[-1]] = gt_masks”上收到此错误,为什么会发生这种情况?跨度>
标签: python-3.x tensorflow object-detection mask valueerror