Skip to content

Commit 4fe4c9c

Browse files
committed
chore(models): update models env flag
1 parent e3f52f9 commit 4fe4c9c

File tree

4 files changed

+5
-12
lines changed

4 files changed

+5
-12
lines changed

e2e/tests/humanseg.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ describe('e2e test humanseg model', () => {
4040
return diffPixelsNum / backImageData.length;
4141
});
4242

43-
const expectedMAP = 0.05;
43+
const expectedMAP = 0.06;
4444
await expect(mAP).toBeLessThanOrEqual(expectedMAP);
4545
});
4646
});

packages/paddlejs-models/gesture/src/index.ts

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,13 +33,12 @@ export async function load() {
3333

3434
detectRunner = new Runner({
3535
modelPath: 'https://paddlejs.bj.bcebos.com/models/fuse/gesture/gesture_det_fuse_activation/model.json',
36-
fill: '#fff'
36+
webglFeedProcess: true
3737
});
3838
const detectInit = detectRunner.init();
3939

4040
recRunner = new Runner({
41-
modelPath: 'https://paddlejs.bj.bcebos.com/models/fuse/gesture/gesture_rec_fuse_activation/model.json',
42-
fill: '#fff'
41+
modelPath: 'https://paddlejs.bj.bcebos.com/models/fuse/gesture/gesture_rec_fuse_activation/model.json'
4342
});
4443
const recInit = recRunner.init();
4544

packages/paddlejs-models/humanseg/src/index.ts

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -34,18 +34,13 @@ export async function load(needPreheat = true, enableLightModel = false, customM
3434
runner = new Runner({
3535
modelPath: path,
3636
needPreheat: needPreheat !== undefined ? needPreheat : true,
37-
feedShape: {
38-
fw: WIDTH,
39-
fh: HEIGHT
40-
},
41-
fill: '#fff',
4237
mean: [0.5, 0.5, 0.5],
43-
std: [0.5, 0.5, 0.5]
38+
std: [0.5, 0.5, 0.5],
39+
webglFeedProcess: true
4440
});
4541

4642
env.set('webgl_pack_channel', true);
4743
env.set('webgl_pack_output', true);
48-
env.set('webgl_feed_process', true);
4944

5045
await runner.init();
5146

packages/paddlejs-models/mobilenet/src/index.ts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,6 @@ export async function load(config: ModelConfig, map: string[] | MobilenetMap) {
3737
scale: 256,
3838
needPreheat
3939
});
40-
env.set('webgl_feed_process', true);
4140
env.set('webgl_pack_channel', true);
4241
await runner.init();
4342
}

0 commit comments

Comments
 (0)