Resaster-50 версия Faster RCNN (обнаружение объектов TensorflowJS) не работает в браузере


1. Я выбрал ResNet-50 версии предварительно обученной модели Faster RCNN (обнаружение объекта)

2. Тренировал эту модель на моих данных (тензор потока ==1.5.0)

  1. Преобразовать замороженную модель (
tensorflowjs_converter  --input_format=tf_frozen_model
--output_node_names='detection_boxes,detection_scores,detection_classes,num_detections'  
    --saved_model_tags=serve frozen_inference_graph.pb  ./web_model

)

  1. Запустить в браузере. Модель работает 3-5 минут и не дает никаких результатов (плохих или хороших)

    async function loadModel() {
                  console.log("model loading..");
                  model = await 
             tf.loadFrozenModel(' https://mmmaks2004.github.io/
                               recognigetensor/tensorflowjs_model.pb',
                              ' https://mmmaks2004.github.io/
                               recognigetensor//weights_manifest.json');
            console.log("model loaded.."); 
            ....

    reader.onload = function (event) { image.onload = function (){ src_canvas.width = image.width; src_canvas.height = image.height; src_ctx.drawImage(image,0,0); var imagedata = src_canvas.toDataURL(); let img = tf.fromPixels(src_canvas).asType("float32").expandDims(); model.executeAsync({image_tensor: img}) .then(function(result){ console.log(result); detection_boxes = result[0].reshape([result[0].size]).dataSync(); detection_scores = result[1].reshape([result[1].size]).dataSync(); detection_classes = result[2].reshape([result[2].size]).dataSync(); num_detections = result[3].reshape([result[3].size]).dataSync(); console.log(num_detections); ...


Что я делаю неправильно?
Моя веб-модель: https://github.com/mmmaks2004/recognigetensor

5. У кого такая конвертированная веб-модель (обнаружение объектов)?
Я хочу попробовать работающую модель. Пожалуйста, помогите мне!

0 ответов

Другие вопросы по тегам