3行代码玩转AI,ml5.js前端机器学习简明指南

今天给大家介绍一个JS库 —— ml5.js。

ml5.js旨在为创意编程提供开箱即用的机器学习算法。该库封装了常用的机器学习算法和预训练模型,基于TensorFlow.js,可单独使用,也可搭配p5.js使用。

官网 https://ml5js.org/

当前版本是0.3.1

有多容易使用?

我们看一下代码:

// Step 1: 使用MobileNet创建图像分类模型
const classifier = ml5.imageClassifier('MobileNet', onModelReady);

// Step 2: 选择一张图片
const img = document.querySelector("#myImage")

// Step 3: 预测图片分类结果
let prediction = classifier.predict(img, gotResults);

// Step 4: 对结果进行操作
function gotResults(err, results) {
  console.log(results);

}

确实非常简单。

目前可以实现哪些功能?

01 图像

imageClassifier 

用于识别图像内容,可用于图像与文字的匹配

poseNet 

用于识别人体姿势的关键点,可用于动作识别

bodyPix、UNET 

用于人体与背景分割,可用于自动抠图

styleTransfer 

风格迁移,可用于图片艺术滤镜

pix2pix、CVAE、DCGAN

图像生成,可用于手绘图生成实景图、根据文本生成图像

SketchRNN

图像生成的另一种方式,可用于基于初始笔画生成图案

YOLO

物体检测,可用于追踪物体

02 声音

soundClassifier

音频分类,可用于语音控制

pitchDetection

音调检测,可用于说话人识别

03 文本

CharRNN

文本生成,可用于输入法联想

Sentiment

情绪预测,可用于文本的情感判断

Word2vec

词转向量,可用于把文本转化为向量,在向量空间中进行计算

04 辅助功能

featureExtractor

图像特征提取,可用于以图搜图、迁移学习

KNNClassifier

K-Nearest Neighbors算法创建分类器,可用于图像分类

17个功能建议一个个进行尝试,练习,掌握。

近期活动

【第8期】上海mixlab2019年中-线下聚会http://hdxu.cn/49xyL

欢迎加入

(0)

相关推荐