Есть ли способ маркировки изображений с помощью нескольких меток в tenorflowjs?
Я уже обучал модель с тензорным потоком и своими собственными метками. Если я теперь предсказываю изображение с этой моделью и тензорным потоком, у меня есть массив с совпадениями. Например. У меня есть изображение с кошкой и собакой. Моя модель будет маркировать 55% собаку и 40% кошку и 5% все остальное. Но мне нравится получать массив меток с ~ 80% Dog и ~ 80% cat.
В противном случае, если я тренирую модель только с двумя метками cat и dog, и мне нравится предсказывать изображение с домом, ятакже получит ярлыки собака и кошка за это. В этом случае мне нравится получать массив без меток
Я использую код из следующего хранилища https://github.com/ADLsourceCode/TensorflowJS/tree/master/Mobilenet_VGG16_Keras_To_TensorflowJS/static