From 3ce8c0d31a4575f3fb541748732750a68fbb7f58 Mon Sep 17 00:00:00 2001 From: Wanming Lin Date: Sun, 5 May 2024 20:10:53 -0700 Subject: [PATCH] Fixed issues introduced by PR-228 (#230) - buildConstantByNpy should have a default vaule for new parameter - Fixed typo in ssd_mobilenetv1_nchw.js --- common/utils.js | 2 +- object_detection/ssd_mobilenetv1_nchw.js | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/common/utils.js b/common/utils.js index 20dbab84..6e87f07a 100644 --- a/common/utils.js +++ b/common/utils.js @@ -84,7 +84,7 @@ export const toHalf = (function() { // Convert npy data in original data type to `targetType`, only support // 'float32' to 'float16' conversion currently. -export async function buildConstantByNpy(builder, url, targetType) { +export async function buildConstantByNpy(builder, url, targetType = 'float32') { const dataTypeMap = new Map([ ['f2', {type: 'float16', array: Uint16Array}], ['f4', {type: 'float32', array: Float32Array}], diff --git a/object_detection/ssd_mobilenetv1_nchw.js b/object_detection/ssd_mobilenetv1_nchw.js index fbbab007..2edb0e14 100644 --- a/object_detection/ssd_mobilenetv1_nchw.js +++ b/object_detection/ssd_mobilenetv1_nchw.js @@ -262,7 +262,7 @@ ${nameArray[1]}_BatchNorm_batchnorm`; if (this.targetDataType_ === 'float16') { boxes = this.builder_.cast(boxes, 'float32'); - scores = this.builder_.cast(boxes, 'float32'); + scores = this.builder_.cast(scores, 'float32'); } return {boxes, scores}; }