turicreate(turicreate-5.0b2)中的对象检测返回空的预测数组

时间:2018-07-10 12:19:59

标签: xcode machine-learning coreml turi createml

您好,我将TuriCreate与目标检测功能结合使用,并使用2个带有60张图像的类标识符,并且像说明中那样成功地训练了数据,但是在对测试数据进行预测时,我得到了空洞的预测。在我的MAC OS Mojave 10.14 Beta和XCode版本10.0 Beta 3上安装了turicreate-5.0b2。这是我使用的代码:

import turicreate as tc
images = tc.load_images('data/images')
annotations = tc.SFrame('data/imagesannotations.csv')
data = images.join(annotations)
train_data, test_data = data.random_split(0.8)
model = tc.object_detector.create(train_data, max_iterations=10)
predictions = model.predict(test_data)

我得到了空的预测输出:

Predicting  1/11
Predicting 11/11
>>> predictions
dtype: list
Rows: 11
[[], [], [], [], [], [], [], [], [], [], []]

请帮助我为什么会得到空洞的预测。我以0.8的比率分割图像的数据集以进行训练和测试,它们彼此相似。

2 个答案:

答案 0 :(得分:0)

这听起来似乎不太可能,但是当使用通过Homebrew安装的Python 2.7时,我也收到了空的预测,无论如何导出模型都会导致崩溃。这里有一个类似的问题:https://github.com/apple/turicreate/issues/80

我卸载了Python,并使用https://www.python.org/downloads/mac-osx/上的安装程序重新安装了2.7.13,并收到了预测。我意识到这是几个月后的事了,所以希望您现在已经解决了您的问题。

答案 1 :(得分:0)

此数据集的10次迭代可能不足以得到可以自信地预测任何内容的模型。结果,所有可能的预测都过于自信,无法满足IOU阈值,因此返回值为空列表。