中文 |
tfjs-yolov3
介绍
完全用js来实现图片中的目标检测
基于yolov3算法和Tensorflow.js库
用tensorflow.js实现yolov3和yolov3-tiny
需要注意的是: 必须是Tensorflow.js@v0.12.4版本以上
特点
可以识别任意尺寸的图片
同时支持yolov3和yolov3-tiny
快速开始
安装
npm install tfjs-yolov3
用法示例
import { yolov3, yolov3Tiny } from 'tfjs-yolov3'
async function start () {
const yolo = await yolov3Tiny() // pre-load model (35M)
// or
// const yolo = await yolov3() // pre-load mode
1