Я использую модель глубокого обучения на основе ImageAI, полученную со следующей страницы github: https://github.com/OlafenwaMoses/Action-Net Когда я передаю файл изображения из своего хранилища в качестве входного пути, он отлично работает. Но когда я передаю данные через веб-камеру с помощью open cv, я получаю следующую ошибку:
Traceback (последний вызов был последним): Файл "/ home / zainzia14 / caffe-master / python / caffe / models / M2048 / live_to_vec_action.py ", строка 129, в прогнозах, вероятности = Forex.predictImage (image_input = im2, input_type = 'array', result_count = 4) Файл" /home/zainzia14/.local/lib/python3 .5 /home/zainzia14/.local/lib/python3.6/site-packages/imageai/Prediction/Custom/custom_utils.py ", строка 37, в decode_predictions each_result.append (CLASS_INDEX [str (i)]) KeyError: '14'
Мой код выглядит так:
...
while time.time() < timeout_start + timeout:
return_value, image1 = camera.read()
# resize image
resized = cv2.resize(im2, dim, interpolation = cv2.INTER_AREA)
print(type(resized))
predictions, probabilities = predictor.predictImage(image_input=im2,input_type='array' ,result_count = 4)
ind = np.argmaprobabilities)
print('Current Label = ',ind)
for prediction, probability in zip(predictions, probabilities):
print(prediction, " : ", probability)'
...