Resaster-50 версия Faster RCNN (обнаружение объектов TensorflowJS) не работает в браузере
1. Я выбрал ResNet-50 версии предварительно обученной модели Faster RCNN (обнаружение объекта)
2. Тренировал эту модель на моих данных (тензор потока ==1.5.0)
- Преобразовать замороженную модель (
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
)
Запустить в браузере. Модель работает 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. У кого такая конвертированная веб-модель (обнаружение объектов)?
Я хочу попробовать работающую модель. Пожалуйста, помогите мне!