我正在尝试使用 MaskRCNN 训练我自己的数据集,我从 https://github.com/matterport/Mask_RCNN 那里获得了代码,我开始使用这个框架训练我的数据集,我得到了这个错误。
我尝试训练其他人的另一个数据集,然后代码运行良好,但不是在我的情况下
Traceback (most recent call last): File "farm.py", line 363, in <module> train(model) File "farm.py", line 183, in train dataset_train.load_custom(args.dataset, "train") File "farm.py", line 135, in load_custom height, width = image.shape[:2] ValueError: not enough values to unpack (expected 2, got 1)
1(. 高度, 宽度 = 图像.形状[:2]
这里的高度和宽度需要 2 个值,我们只提供 1 个值,在这种情况下,您将获得以下错误:-
值错误:没有足够的值来解压缩(预期为 2,得到 1(
2(. 打印图像形状[0:2] 的值以查看值的数量。
print(image.shape[0:2])
检查print(image.shape[0:2])
你会知道你错在哪里。