Запуск предварительно обученной модели ONNX — распознавание изображенийPython

Программы на Python
Anonymous
Запуск предварительно обученной модели ONNX — распознавание изображений

Сообщение Anonymous »

Я пытаюсь запустить предварительно обученную модель ONNX (обученную с помощью стороннего инструмента маркировки) для распознавания изображений. Модель обучается с помощью некоторых заранее определенных меток в инструменте. Следующая цель теперь — иметь возможность запускать эту модель вне инструмента. Для этого я беру образец изображения и пытаюсь запустить его через модель, чтобы получить идентифицированные метки в качестве вывода. При этом я столкнулся с препятствием относительно настройки входов. Для модели требуются следующие входные данные:
Изображение

Как могу ли я настроить вводимые данные в следующем коде?

Код: Выделить всё

import cv2
import numpy as np
import onnxruntime
import pytesseract
import PyPDF2

# Load the image
image = cv2.imread("example.jpg")

# Check if the image has been loaded successfully
if image is None:
raise ValueError("Failed to load the image")

# Get the shape of the image
height, width = image.shape[:2]

# Make sure the height and width are positive
if height 

Подробнее здесь: [url]https://stackoverflow.com/questions/75360420/running-a-pre-trained-onnx-model-image-recognition[/url]

Вернуться в «Python»