Skip to content

Commit 2cb332d

Browse files
stick open pose model to opset7 (#3055)
1 parent 2543996 commit 2cb332d

File tree

1 file changed

+2
-1
lines changed
  • demos/common/python/openvino/model_zoo/model_api/models

1 file changed

+2
-1
lines changed

demos/common/python/openvino/model_zoo/model_api/models/open_pose.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
import cv2
1818
import ngraph as ng
19+
import ngraph.opset7 as opset7
1920
import numpy as np
2021
try:
2122
from numpy.core.umath import clip
@@ -60,7 +61,7 @@ def __init__(self, model_adapter, configuration=None, preload=False):
6061
# Heuristic NMS kernel size adjustment depending on the feature maps upsampling ratio.
6162
p = int(np.round(6 / 7 * self.upsample_ratio))
6263
k = 2 * p + 1
63-
pooled_heatmap = ng.max_pool(heatmap, kernel_shape=(k, k), pads_begin=(p, p), pads_end=(p, p),
64+
pooled_heatmap = opset7.max_pool(heatmap, kernel_shape=(k, k), pads_begin=(p, p), pads_end=(p, p),
6465
strides=(1, 1), name=self.pooled_heatmaps_blob_name)
6566
f = ng.impl.Function(
6667
[ng.result(heatmap, name=self.heatmaps_blob_name),

0 commit comments

Comments
 (0)