用web端实现抠图,以及大模型本地化部署遇到坑
介绍
图像识别相信大家都不陌生了,而比较成熟的一些大模型算法都是交给了后端来处理。后端需要通过网络传输,没法做到实时处理。所以前端web也是需要有一定的运算能力才行。比如摄像头了出现了人,你这边想实时的观看这个人的位置,并标记出来是谁等。我们这边暂时以人物的抠图为例,去除背景。这里我遇到了一个最大的坑就是模型的本地部署,这边总是部署失败。我找了三个小时才找到原因。后续如果大家不想也走那么多弯路,就跟着我细细道来。抠图效果

步骤
安装插件
这里的tensorflow也是模型,相当于一个模型分析器,可以分析你本地或线上的模型。
而body-pix插件也是处理body-pix模型用的。
import * as bodyPix from '@tensorflow-models/body-pix'
加载模型
这里的坑实在太多了,这里有个modelUrl其实就是加载模型的地址,我们可以随意更换,不用线上的模型的意思,如果你不填写则会自动用线上的模型。
let n