Commit 3f8d0bc3 authored by jay's avatar jay
Browse files

Initial subpixel stubbing

parent 195ffba8

autocnet/matcher/subpixel.py

+43 −0
Original line number Diff line number Diff line 
import numpy as np



from skimage.feature import register_translation



from autocnet.matcher import naive_template

from autocnet.matcher import ciratefi
@@ -55,6 +57,47 @@ def clip_roi(img, center, img_size): 
                                             x_stop + 1, y_stop + 1])

    return clipped_img



def subpixel_phase(template, search, **kwargs):

    """

    Apply the spectral domain matcher to a search and template image. To

    shift the images, the x_shift and y_shift, need to be subtracted from

    the center of the search image. It may also be necessary to apply the

    fractional pixel adjustment as well (if for example the center of the

    search is not an integer); this function do not manage shifting.



    Parameters

    ----------

    template : ndarray

               The template used to search



    search : ndarray

             The search image



    Returns

    -------

    x_offset : float

               Shift in the x-dimension



    y_offset : float

               Shift in the y-dimension



    strength : tuple

               With the RMSE error and absolute difference in phase

    """

    if not template.shape == search.shape:

        raise ValueError('Both the template and search images must be the same shape.')



    (y_shift, x_shift), error, diffphase = register_translation(search, template, **kwargs)

    return x_shift, y_shift, (error, diffphase)



def subpixel_composite(asub, bsub, b_subpixel_shift=[0,0], template_extents=25, phase_kwargs={}, template_kwargs={}):

    x_shift, y_shift, error = subpixel_phase(asub, bsub, **phase_kwargs)



    asize = asub.shape

    bsize = bsub.shape



    amini = asub[asub.shape]





def subpixel_offset(template, search, **kwargs):

    """

conda/meta.yaml

+2 −1
Original line number Diff line number Diff line 
package:

  name: autocnet

  version: 0.2.0

  version: 0.2.2

  

channels:

  - conda-forge
@@ -42,6 +42,7 @@ requirements: 
    - numpy

    - plio

    - gdal

    - scikit-image



test:

  imports:

setup.py

+1 −3
Original line number Diff line number Diff line
@@ -7,8 +7,6 @@ with open('README.rst', 'r') as f: 
    long_description = f.read()





VERSION = autocnet.__version__



def setup_package():

    examples = set()

    for i in available():
@@ -23,7 +21,7 @@ def setup_package(): 


    setup(

        name = "autocnet",

        version = VERSION,

        version = '0.2.2',

        author = "Jay Laura",

        author_email = "jlaura@usgs.gov",

        description = ("I/O API to support planetary data formats."),