Skip to content

Commit a353d5e

Browse files
authored
Merge pull request #384 from JingyuanZhang/master
fix(core): fix feedShape in modelConfig webglFeedProcess
2 parents 4f6cba0 + 779ec71 commit a353d5e

File tree

5 files changed

+20
-18
lines changed

5 files changed

+20
-18
lines changed

packages/paddlejs-core/package.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
{
22
"name": "@paddlejs/paddlejs-core",
3-
"version": "2.1.14",
3+
"version": "2.1.15",
44
"description": "",
55
"main": "lib/index",
66
"scripts": {

packages/paddlejs-core/src/graph.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ export default class ModelGraph {
3838
this.type = config.type || this.type;
3939
this.plugins = config.plugins;
4040
this.config = config;
41+
model.feedShape && (this.config.feedShape = model.feedShape);
4142
}
4243

4344
/**

packages/paddlejs-models/ocr/package-lock.json

Lines changed: 13 additions & 13 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

packages/paddlejs-models/ocr/package.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,9 @@
1010
"author": "",
1111
"license": "ISC",
1212
"dependencies": {
13+
"@paddlejs-mediapipe/opencv": "0.0.3",
1314
"@paddlejs/paddlejs-backend-webgl": "^1.1.15",
14-
"@paddlejs/paddlejs-core": "^2.1.14",
15-
"@paddlejs-mediapipe/opencv": "0.0.3"
15+
"@paddlejs/paddlejs-core": "^2.1.15"
1616
},
1717
"devDependencies": {
1818
"d3-polygon": "^2.0.0",

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,13 +59,13 @@ export async function init(detCustomModel = null, recCustomModel = null) {
5959
const detModelPath = 'https://paddlejs.bj.bcebos.com/models/fuse/ocr/ch_PP-OCRv2_det_fuse_activation/model.json';
6060
const recModelPath = 'https://paddlejs.bj.bcebos.com/models/fuse/ocr/ch_PP-OCRv2_rec_fuse_activation/model.json';
6161
env.set('webgl_pack_output', true);
62-
env.set('webgl_feed_process', true);
6362
detectRunner = new Runner({
6463
modelPath: detCustomModel ? detCustomModel : detModelPath,
6564
fill: '#fff',
6665
mean: [0.485, 0.456, 0.406],
6766
std: [0.229, 0.224, 0.225],
68-
bgr: true
67+
bgr: true,
68+
webglFeedProcess: true
6969
});
7070
const detectInit = detectRunner.init();
7171

@@ -75,6 +75,7 @@ export async function init(detCustomModel = null, recCustomModel = null) {
7575
mean: [0.5, 0.5, 0.5],
7676
std: [0.5, 0.5, 0.5],
7777
bgr: true,
78+
webglFeedProcess: true,
7879
plugins: {
7980
preTransforms: [new OptModel()]
8081
}

0 commit comments

Comments
 (0)