Блог

Я пытаюсь выполнить вывод классификации изображений на модели CNN (model.tflite) с помощью (image_test.jpg) но это не работает. Как я могу перейти с HxWxC на CxHxW ?

 # Load the TFLite model and allocate tensors
interpreter = tf.lite.Interpreter(model_path="model.tflite")
interpreter.allocate_tensors()

# Get input and output tensors
input_details = interpreter.get_input_details()
print(input_details)
output_details = interpreter.get_output_details()


path = 'image_test.jpg'
img = cv2.imread(path)
new_img = cv2.resize(img, (224, 224))
new_img = np.array(new_img, dtype=np.float32)
# input_details[0]['index'] = the index which accepts the input
# expected type FLOAT32 for input 0, name: serving_default_input_1 

interpreter.set_tensor(input_details[0]['index'], [new_img])

# run the inference
interpreter.invoke()

# output_details[0]['index'] = the index which provides the input
output_data = interpreter.get_tensor(output_details[0]['index'])

print("For file {}, the output is {}".format(file.stem, output_data))
 

Я получаю эту ошибку :

 --
ValueError                                Traceback (most recent call last)
<ipython-input-140-8dbf21f6d8dd> in <module>()
      9 # new_img = np.expand_dims(new_img, axis=0)
     10 
---> 11 interpreter.set_tensor(input_details[0]['index'], [new_img])
     12 
     13 # run the inference

/usr/local/lib/python3.7/dist-packages/tensorflow/lite/python/interpreter.py in set_tensor(self, tensor_index, value)
    605       ValueError: If the interpreter could not set the tensor.
    606     """
--> 607     self._interpreter.SetTensor(tensor_index, value)
    608 
    609   def resize_tensor_input(self, input_index, tensor_size, strict=False):

ValueError: Cannot set tensor: Dimension mismatch. Got 224 but expected 3 for dimension 1 of input 0.
 

Ценю любую помощь. Заранее благодарю вас!