Есть ли способ использовать пользовательские модели в ml5 yolo ()? - PullRequest
0 голосов
/ 27 марта 2019

с мл5 yolo() упомяните, что

"эта реализация в значительной степени является производной от ModelDepot".

В нем не упоминалось, какую модель подготовки к поезду она использует или как вы можете использовать собственную обученную модель

let video;
let yolo;
let status;
let objects = [];

function setup() {
  createCanvas(320, 240);
  video = createCapture(VIDEO);
  video.size(320, 240);

  // Create a YOLO method
  yolo = ml5.YOLO(video, startDetecting);

  // Hide the original video
  video.hide();
  status = select('#status');
}

function draw() {
  image(video, 0, 0, width, height);
  for (let i = 0; i < objects.length; i++) {
    noStroke();
    fill(0, 255, 0);
    text(objects[i].className, objects[i].x * width, objects[i].y * height - 5);
    noFill();
    strokeWeight(4);
    stroke(0, 255, 0);
    rect(objects[i].x * width, objects[i].y * height, objects[i].w * width, objects[i].h * height);
  }
}

function startDetecting() {
  status.html('Model loaded!');
  detect();
}

function detect() {
  yolo.detect(function(err, results) {
    objects = results;
    detect();
  });
}

1 Ответ

0 голосов
/ 09 июня 2019

Я читал документацию по yolo и нашел способ:

ml5-yolo-library -> Сильно получен из https://github.com/ModelDepot/tfjs-yolo-tiny (ModelDepot: modeldepot.io)

здесь -> https://modeldepot.io/mikeshi/tiny-yolo-in-javascript

говорит -> Эта модель была создана из взятых оригинальных cfg и весов Darknet Tiny YOLO, преобразования их в Keras через YAD2K, а затем с помощью tenorflowjs_converter для преобразования в Tensorflow.формат js.

Итак, я пытаюсь сделать то же самое, что и вы прямо сейчас.надеюсь, я найду способ

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...