Преобразование 3D-тензора обратно в изображение (jpg / png)

#react-native #tensorflow #tensorflow.js

#react-native #tensorflow #tensorflow.js

Вопрос:

Я хочу использовать компонент камеры Tensorflow, который возвращает Tensor3D для каждого кадра предварительного просмотра камеры. Это отлично подходит для живой классификации с помощью моделей tensorflow, однако я хочу иметь возможность конвертировать этот тензор обратно в формат jpg или изображения для использования с библиотекой распознавания текста, такой как Tesseract.

Есть ли какой-либо способ преобразовать <tf.Tensor3d> в более традиционный формат изображения для общего использования?

Ответ №1:

Вы можете использовать tf.node.encodeJpeg или tf.node.encodePng, но для них требуется, чтобы передаваемый 3D-тензор имел тип uint8 .

Комментарии:

1. Есть ли способ преобразовать возвращаемый int32 из imageTensor в uint8?

2. Спасибо за это, я видел, но, к сожалению, мой тензор находится в int32, я сделал запрос функции на странице tensorflow, так что, надеюсь, они смогут расширить доступные входные данные

3. Я думал tf.cast , что это сработает, но тогда он поддерживает только приведение к int32 .