I trained the object detection model into 3 classes using CreateML. Loaded the model into coremltools and performed the prediction, got 'coordinates' and 'confidence' but got no class labels. How can I get class labels?
{'coordinates': array([[0.31552333, 0.30995899, 0.38225624, 0.59993058],
[0.41404313, 0.5812282 , 0.07037259, 0.06291649],
[0.23581643, 0.08297428, 0.18884215, 0.12770388],
[0.42899013, 0.18289472, 0.06921174, 0.06600466]]), 'confidence': array([[9.92818241e-06, 9.99987483e-01, 8.74077966e-09],
[1.00150883e-05, 1.29828220e-06, 9.98730123e-01],
[9.98335183e-01, 1.02320407e-03, 2.00047023e-09],
[6.40532759e-04, 1.93859320e-04, 8.27081084e-01]])}
def get_feature_size(spec):
		input_description = spec.description.input[0]
		inputSizeImg = input_description.type.imageType.imageSizeRange.widthRange.lowerBound, \
									 input_description.type.imageType.imageSizeRange.heightRange.lowerBound
		return inputSizeImg
loaded_model = coremltools.models.MLModel(modelPath)
inputSize_img = get_feature_size(loaded_model.get_spec())
with Image.open(imgPath).resize(input_size_img) as img:
out_dict = model.predict({'image':img})