From 6d9f0889752d2f8e0de7f3968da280cdf5e95f4a Mon Sep 17 00:00:00 2001 From: Lijun Zhu Date: Mon, 30 Nov 2020 10:30:23 -0800 Subject: [PATCH] PyCuAmpcor: cuDenseOffsets.py towork with input varying gross offsets --- contrib/PyCuAmpcor/examples/cuDenseOffsets.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/contrib/PyCuAmpcor/examples/cuDenseOffsets.py b/contrib/PyCuAmpcor/examples/cuDenseOffsets.py index 48ad838..5ee271d 100755 --- a/contrib/PyCuAmpcor/examples/cuDenseOffsets.py +++ b/contrib/PyCuAmpcor/examples/cuDenseOffsets.py @@ -268,8 +268,14 @@ def estimateOffsetField(reference, secondary, inps=None): if grossOffset.size != 2*numberWindows : print('The input gross offsets do not match the number of windows {} by {} in int32 type'.format(objOffset.numberWindowDown, objOffset.numberWindowAcross)) return 0; - grossOffset.reshape(numberWindows, 2) - objOffset.setVaryingGrossOffset(grossOffset[:,0], grossOffset[:,1]) + grossOffset = grossOffset.reshape(numberWindows, 2) + grossAzimuthOffset = grossOffset[:, 0] + grossRangeOffset = grossOffset[:, 1] + # enforce C-contiguous flag + grossAzimuthOffset = grossAzimuthOffset.copy(order='C') + grossRangeOffset = grossRangeOffset.copy(order='C') + # set varying gross offset + objOffset.setVaryingGrossOffset(grossAzimuthOffset, grossRangeOffset) # check objOffset.checkPixelInImageRange()