Question

我创建了一个sagemaker对象检测培训工作以及随后的端点。我只有两节课。但是，当我尝试使用端点进行预测时，响应中会出现多行，如下所示：

{"prediction": [
[1.0, 0.632090151309967, 0.0, 0.003549516201019287, 1.0, 1.0], 
[0.0, 0.4135304093360901, 0.0, 0.006693154573440552, 1.0, 0.9729366302490234], 
[0.0, 0.018929673358798027, 0.0, 0.044431887567043304, 0.6495294570922852, 0.23290297389030457], 
[0.0, 0.01802791841328144, 0.0, 0.11557215452194214, 0.6625108122825623, 0.7412691712379456], 
[0.0, 0.015324527397751808, 0.0, 0.267954021692276, 0.6784608960151672, 0.39592066407203674], 
[0.0, 0.013910820707678795, 0.0, 0.8590829372406006, 0.7399784326553345, 1.0], 
[1.0, 0.013243389315903187, 0.928236186504364, 0.0, 1.0, 0.07348344475030899], 
[0.0, 0.012794392183423042, 0.9662157893180847, 0.0, 1.0, 0.057823698967695236], 
[0.0, 0.011968772858381271, 0.0, 0.9265779256820679, 0.04517384618520737, 1.0], 
[0.0, 0.011287822388112545, 0.953392744064331, 0.9526442885398865, 1.0, 1.0], 
[1.0, 0.01005781814455986, 0.8989022970199585, 0.9481537342071533, 1.0, 1.0]
]
}

为什么响应中有多行？

Answer 1

每行对应于一个检测到的对象。引用https://docs.aws.amazon.com/sagemaker/latest/dg/object-detection-in-formats.html：

.json文件中的每一行都包含一个表示检测到的对象的数组。这些对象数组中的每一个都包含六个数字的列表。第一个数字是预测的类别标签。第二个数字是与检测相关的置信度得分。最后四个数字表示边界框坐标[xmin，ymin，xmax，ymax]。

Sagemaker对象检测预测

1 个答案: