Пример #1
0
def webGLBrainMidSaggitalGaussianExample():

    basePath = 'images/webGLMid.png'
    targetPath = 'images/webGLwarpedMid.png'

    base = readImage(basePath)
    target = readImage(targetPath)

    print(base.shape)

    start = time.time()

    #deformParams = (20, 3, [-50, 50, -50, 50], 'gaussian')
    deformParams = (30, 3, [], 'gaussian')

    (p, rIm) = TaylorMSConvexRegistration(target, base, deformParams, 1)

    print('Total Time elapsed: ', (time.time() - start) / 60, ' minutes\n\n')

    msDiff = meanSquaresMetric3D(target, rIm)

    print('Deformation Coefficients: ', p, '\n\n')
    print('Registration Parameters: ', deformParams, '\n\n')
    print('Mean Squares Difference: ', msDiff, '\n\n')

    pilIm = Image.fromarray(np.uint8(rIm))
    pilIm.save(
        '/Users/brknight/Documents/ConvexOptimization/figures/webGLMidEx.png')
Пример #2
0
def DeformedBrainPDExample():

    basePath = 'images/BrainProtonDensitySlice.png'
    targetPath = 'images/DeformedBrainPD.png'

    base = readImage(basePath)
    target = readImage(targetPath)

    start = time.time()

    # deformParams = (30, 3, [-50, 50, -50, 50], 'gaussian')
    # 	deformParams = (27, 3, [-50, 50, -50, 50], 'gaussian') current registration parameters
    deformParams = (35, 3, [], 'gaussian')
    (p, rIm) = TaylorMSConvexRegistration(target, base, deformParams, 1)

    print('Total Time elapsed: ', (time.time() - start) / 60, ' minutes\n\n')

    msDiff = meanSquaresMetric3D(target, rIm)

    print('Deformation Coefficients: ', p, '\n\n')
    print('Registration Parameters: ', deformParams, '\n\n')
    print('Mean Squares Difference: ', msDiff, '\n\n')

    pilIm = Image.fromarray(np.uint8(rIm))
    # pilIm.save('/Users/brknight/Documents/ConvexOptimization/figures/DeformedBrainPDEx.png')
    pilIm.save(
        '/Users/brknight/Documents/ConvexOptimization/figures/DeformedBrainPDUnboundedExTest.png'
    )
Пример #3
0
def WarpedRiceCVXExample():

    basePath = 'images/rice.png'
    targetPath = 'images/WarpedRice.png'

    base = readImage(basePath, (300, 300))
    target = readImage(targetPath)

    start = time.time()

    deformParams = (50, 3, [-50, 50, -50, 50], 'gaussian')
    (p, rIm) = cvxTaylorMSConvexRegistration(target, base, deformParams, 1)

    print('Total Time elapsed: ', (time.time() - start) / 60, ' minutes\n\n')

    msDiff = meanSquaresMetric3D(target, rIm)

    print('Deformation Coefficients: ', p, '\n\n')
    print('Registration Parameters: ', deformParams, '\n\n')
    print('Mean Squares Difference: ', msDiff, '\n\n')

    pilIm = Image.fromarray(np.uint8(rIm))
    pilIm.save(
        '/Users/brknight/Documents/ConvexOptimization/figures/WarpedRiceCVXEx.png'
    )
Пример #4
0
def BrainPDShiftedX13Y17Example():

    basePath = 'images/BrainProtonDensitySlice.png'
    targetPath = 'images/BrainProtonDensitySliceShifted13x17y.png'

    base = readImage(basePath)
    target = readImage(targetPath)

    start = time.time()

    deformParams = (30, 3, [], 'firstOrder')
    (p, rIm) = TaylorMSConvexRegistration(target, base, deformParams, 1)

    print('Total Time elapsed: ', (time.time() - start) / 60, ' minutes\n\n')

    msDiff = meanSquaresMetric3D(target, rIm)

    print('Deformation Coefficients: ', p, '\n\n')
    print('Registration Parameters: ', deformParams, '\n\n')
    print('Mean Squares Difference: ', msDiff, '\n\n')

    pilIm = Image.fromarray(np.uint8(rIm))
    pilIm.save(
        '/Users/brknight/Documents/ConvexOptimization/figures/BrainPDShiftedX13Y17Ex2.png'
    )
Пример #5
0
def BrainMidSaggitalThirdOrderExample():

    basePath = 'images/BrainMidSagittalSlice.png'
    targetPath = 'images/DefMid.jpg'

    base = readImage(basePath, (187, 155))
    target = readImage(targetPath)

    start = time.time()

    deformParams = (30, 3, [-50, 50, -50, 50], 'thirdOrder')
    (p, rIm) = cvxTaylorMSConvexRegistration(target, base, deformParams, 1)

    print('Total Time elapsed: ', (time.time() - start) / 60, ' minutes\n\n')

    msDiff = meanSquaresMetric3D(target, rIm)

    print('Deformation Coefficients: ', p, '\n\n')
    print('Registration Parameters: ', deformParams, '\n\n')
    print('Mean Squares Difference: ', msDiff, '\n\n')

    pilIm = Image.fromarray(np.uint8(rIm))
    pilIm.save(
        '/Users/brknight/Documents/ConvexOptimization/figures/BrainMidSaggitalThirdOrderEx.png'
    )
Пример #6
0
def MiddleburryExample():
    basePath = 'images/barn1/im0.ppm'
    targetPath = 'images/barn1/im8.ppm'

    base = readImage(basePath)
    target = readImage(targetPath)

    start = time.time()

    deformParams = (20, 3, [-50, 50, -50, 50], 'firstOrder')
    (p, rIm) = TaylorMSConvexRegistration(target, base, deformParams, 1)

    print('Total Time elapsed: ', (time.time() - start) / 60, ' minutes\n\n')

    msDiff = meanSquaresMetric3D(target, rIm)

    print('Deformation Coefficients: ', p, '\n\n')
    print('Registration Parameters: ', deformParams, '\n\n')
    print('Mean Squares Difference: ', msDiff, '\n\n')

    pilIm = Image.fromarray(np.uint8(rIm))
    pilIm.save(
        '/Users/brknight/Documents/ConvexOptimization/figures/MiddleburryEx.png'
    )