image-processing - 使用 turicreate 对大量高分辨率图片进行训练

Question

我正在尝试使用 Turicreate 在大约 150 张非常高分辨率的图片（每张 4Mb，3000X5000）上训练模型。我在跑

model = tc.object_detector.create(train_data, max_iterations=10)

过了一会儿，我收到“低虚拟内存”警告，并且在我的计算机卡住之后。

我想知道在这里能够训练这样一批图片的最佳做法是什么。

我正在使用的完整代码：

import turicreate as tc
data =  tc.SFrame('annotations.sframe')
train_data, test_data = data.random_split(0.8)
model = tc.object_detector.create(train_data, max_iterations=10)
predictions = model.predict(test_data)
metrics = model.evaluate(test_data)
model.save('mymodel.model')
model.export_coreml('MyCustomObjectDetector.mlmodel')

标签： image-processingcomputer-visionturi-create

Answer 1

将数据集图像的大小例如减小到（宽度：400 高度：300），并将 max_iterations 提高到至少 1000。