今天给大家介绍一个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 图像
styleTransfer
风格迁移,可用于图片艺术滤镜
pix2pix、CVAE、DCGAN
图像生成,可用于手绘图生成实景图、根据文本生成图像
SketchRNN
图像生成的另一种方式,可用于基于初始笔画生成图案
YOLO
物体检测,可用于追踪物体
02 声音
soundClassifier
音频分类,可用于语音控制
音调检测,可用于说话人识别
03 文本
文本生成,可用于输入法联想
Sentiment
情绪预测,可用于文本的情感判断
Word2vec
词转向量,可用于把文本转化为向量,在向量空间中进行计算
04 辅助功能
图像特征提取,可用于以图搜图、迁移学习
K-Nearest Neighbors算法创建分类器,可用于图像分类
17个功能建议一个个进行尝试,练习,掌握。
近期活动