Сегментное изображение с моделью Deeplab Tensor Flow React Native - PullRequest
Новая
       56

Сегментное изображение с моделью Deeplab Tensor Flow React Native

0 голосов
/ 20 апреля 2020

Я пытаюсь использовать предтренированную модель Deeplab из https://www.npmjs.com/package/@tensorflow-models / deeplab для обеспечения сегментации изображения.

Я настроил и работаю https://github.com/tensorflow/tfjs/tree/master/tfjs-react-native

Но когда я пытаюсь передать Tensor в модель, я получаю следующую ошибку:

Ошибка

"Dtype of dict ['ImageTensor'] в файле model.execute (dict) должно быть указано значение int32, но оно имеет тип float32 "

Я пробовал использовать оба типа dtypes в тензорном int32 (imageTensor) и float32 (floatImageTensor), оба получили одинаковую ошибку. Любая помощь приветствуется.

Мой код выглядит так: 

    const response = await fetch(imagePath, {}, {isBinary: true});

    const rawImageData = await response.arrayBuffer();

    const uint8View = new Uint8Array(rawImageData);

    const imageTensor = decodeJpeg(uint8View);

    const floatImageTensor = tf.cast(imageTensor, 'float32');

    console.log(imageTensor, floatImageTensor);

    const res = await tfModel.segment(imageTensor);
    return res;
  };
...