Есть потребность запустить nsfw классификатор на ios и android устройствах. Приложение пишется на react-native.
Используя react-native-tensorflow пришел к вот такому результату (в изображении ниже). Предварительный поиск в google подсказывает, что модель тренировалась на версии tensorflow не совпадающей с версией в react-native-tensorflow. Используемый код:
const tfImageRecognition = new TfImageRecognition({
model:require('./assets/nsfw.299x299.pb'),
labels: require('./assets/nsfw_labels.txt')
})
const results = await tfImageRecognition.recognize({
image: this.image,
inputName: "input_1", //Optional, defaults to "input"
outputName: "dense_3/Softmax", //Optional, defaults to "output"
})
const resultText = `Name: ${results[0].name} - Confidence: ${results[0].confidence}`
this.setState({result: resultText})
Есть ли способ пересобрать модель для нужной версии?
Если у кого-то получалось запускать подобные классификаторы изображений на react-native поделитесь пожалуйста опытом. Рассматриваются все возможные варианты, например использование tensorflow/tfjs. У меня правда не получилось приспособить tensorflow/tfjs к мобильным устройствам, хотя в браузере все прекрасно работает.
Виртуальный выделенный сервер (VDS) становится отличным выбором