Commit ebda2589 by Zhao Wu Committed by Tianqi Chen

[CI] Solve occasional CI issue when pad value is all 0 (#3801)

parent 3c8901ad
...@@ -74,6 +74,7 @@ def _ConvolutionLayerParams(op, inexpr, etab): ...@@ -74,6 +74,7 @@ def _ConvolutionLayerParams(op, inexpr, etab):
pad_l = valid.paddingAmounts.borderAmounts[1].startEdgeSize pad_l = valid.paddingAmounts.borderAmounts[1].startEdgeSize
pad_b = valid.paddingAmounts.borderAmounts[0].endEdgeSize pad_b = valid.paddingAmounts.borderAmounts[0].endEdgeSize
pad_r = valid.paddingAmounts.borderAmounts[1].endEdgeSize pad_r = valid.paddingAmounts.borderAmounts[1].endEdgeSize
if not all(v == 0 for v in (pad_t, pad_l, pad_b, pad_r)):
inexpr = _op.nn.pad(data=inexpr, pad_width=((0, 0), inexpr = _op.nn.pad(data=inexpr, pad_width=((0, 0),
(0, 0), (0, 0),
(pad_t, pad_b), (pad_t, pad_b),
...@@ -208,6 +209,7 @@ def _PoolingLayerParams(op, inexpr, etab): ...@@ -208,6 +209,7 @@ def _PoolingLayerParams(op, inexpr, etab):
pad_l = valid.paddingAmounts.borderAmounts[1].startEdgeSize pad_l = valid.paddingAmounts.borderAmounts[1].startEdgeSize
pad_b = valid.paddingAmounts.borderAmounts[0].endEdgeSize pad_b = valid.paddingAmounts.borderAmounts[0].endEdgeSize
pad_r = valid.paddingAmounts.borderAmounts[1].endEdgeSize pad_r = valid.paddingAmounts.borderAmounts[1].endEdgeSize
if not all(v == 0 for v in (pad_t, pad_l, pad_b, pad_r)):
params['padding'] = [pad_t, pad_l, pad_b, pad_r] params['padding'] = [pad_t, pad_l, pad_b, pad_r]
elif op.WhichOneof('PoolingPaddingType') == 'includeLastPixel': elif op.WhichOneof('PoolingPaddingType') == 'includeLastPixel':
# I don't know if this is correct # I don't know if this is correct
......
...@@ -20,5 +20,7 @@ def get_cat_image(): ...@@ -20,5 +20,7 @@ def get_cat_image():
dst = 'cat.png' dst = 'cat.png'
real_dst = download_testdata(url, dst, module='data') real_dst = download_testdata(url, dst, module='data')
img = Image.open(real_dst).resize((224, 224)) img = Image.open(real_dst).resize((224, 224))
img = np.transpose(img, (2, 0, 1))[np.newaxis, :] # CoreML's standard model image format is BGR
img_bgr = np.array(img)[:, :, ::-1]
img = np.transpose(img_bgr, (2, 0, 1))[np.newaxis, :]
return np.asarray(img) return np.asarray(img)
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment