56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
#
|
||
|
|
# test_cuAmpcor.py
|
||
|
|
# Test program to run ampcor with GPU
|
||
|
|
#
|
||
|
|
#
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import numpy as np
|
||
|
|
#from PyCuAmpcor import PyCuAmpcor
|
||
|
|
from isce.components.contrib.PyCuAmpcor import PyCuAmpcor
|
||
|
|
|
||
|
|
def main():
|
||
|
|
'''
|
||
|
|
main program
|
||
|
|
'''
|
||
|
|
|
||
|
|
objOffset = PyCuAmpcor()
|
||
|
|
|
||
|
|
objOffset.algorithm = 0
|
||
|
|
objOffset.deviceID = 0 # -1:let system find the best GPU
|
||
|
|
objOffset.nStreams = 2 #cudaStreams
|
||
|
|
objOffset.masterImageName = "master.slc"
|
||
|
|
objOffset.masterImageHeight = 43008
|
||
|
|
objOffset.masterImageWidth = 24320
|
||
|
|
objOffset.slaveImageName = "slave.slc"
|
||
|
|
objOffset.slaveImageHeight = 43008
|
||
|
|
objOffset.slaveImageWidth = 24320
|
||
|
|
objOffset.windowSizeWidth = 64
|
||
|
|
objOffset.windowSizeHeight = 64
|
||
|
|
objOffset.halfSearchRangeDown = 20
|
||
|
|
objOffset.halfSearchRangeAcross = 20
|
||
|
|
objOffset.derampMethod = 1
|
||
|
|
objOffset.numberWindowDown = 300
|
||
|
|
objOffset.numberWindowAcross = 30
|
||
|
|
objOffset.skipSampleDown = 128
|
||
|
|
objOffset.skipSampleAcross = 64
|
||
|
|
objOffset.numberWindowDownInChunk = 10
|
||
|
|
objOffset.numberWindowAcrossInChunk = 10
|
||
|
|
objOffset.corrSurfaceOverSamplingFactor = 8
|
||
|
|
objOffset.corrSurfaceZoomInWindow = 16
|
||
|
|
objOffset.corrSufaceOverSamplingMethod = 1
|
||
|
|
objOffset.mmapSize = 8
|
||
|
|
|
||
|
|
objOffset.setupParams()
|
||
|
|
objOffset.masterStartPixelDownStatic = 1000
|
||
|
|
objOffset.masterStartPixelAcrossStatic = 1000
|
||
|
|
objOffset.setConstantGrossOffset(642, -30)
|
||
|
|
objOffset.checkPixelInImageRange()
|
||
|
|
objOffset.runAmpcor()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
|
||
|
|
main()
|