Skip to content

Commit

Permalink
Update input sizes to calculate padding for maxPool2d of TinyYolo nch…
Browse files Browse the repository at this point in the history
  • Loading branch information
BruceDai authored and Honry committed May 15, 2024
1 parent 2d60e29 commit 444fd57
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion object_detection/tiny_yolov2_nchw.js
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ export class TinyYoloV2Nchw {

buildMaxPool2d_(input, options) {
options.padding = computePadding2DForAutoPad(
/* nhwc */[input.shape()[1], input.shape()[2]],
/* nchw */[input.shape()[2], input.shape()[3]],
options.windowDimensions,
options.strides, options.dilations, 'same-upper');
return this.builder_.maxPool2d(input, options);
Expand Down

0 comments on commit 444fd57

Please sign in to comment.