natelust / least_asymmetry Goto Github PK

def fit_least_asymmetry_centering(imageCube, yguess, xguess, asym_rad):
    '''
        imageCube : nD-array (assume: (60k, 32, 32))
        yguess    : initial y center guess (assume: 15)
        xguess    : initial x center guess (assume: 15)
        asym_rad  : width of asymmetry subframe -- attempt to control corner cases (assume: 5)
    '''
    y,x = 0,1
    
    nFrames = imageCube.shape[0]
    
    yinds0, xinds0 = indices(imageCube[0].shape)
    
    nAsymParams = 2 # Xc, Yc
    centering_LeastAsym  = np.zeros((nFrames, nAsymParams))

    for kframe in tqdm(range(nFrames), total=nFrames):
        center_asym = least_asymmetry.actr(imageCube[kframe], [yguess, xguess], \
                           asym_rad=asym_rad, asym_size=5, maxcounts=2, method='gaus', \
                           half_pix=False, resize=False, weights=False)[0]

        try:
            centering_LeastAsym[kframe]  = center_asym[::-1]
        except:
            print('Least Asymmetry FAILED: Setting centering_LeastAsym[%s] to Initial Guess: [%s,%s]' % (kframe, yguess, xguess))
            centering_LeastAsym[kframe]  = np.arange([yguess, xguess])

    return centering_LeastAsym

from sklearn.externals import joblib
from pylab import *;ion()
import least_asymmetry

xgf_imageCube = joblib.load('xgf_image_cube_array.pickle') # input 60k x 32 x 32 nD-array

yguess   = 15
xguess   = 15
asym_rad = 5 # tried 8 before too

fit_least_asymmetry_centering(xgf_imageCube, yguess, xguess, asym_rad)

Traceback (most recent call last):
  File "least_asym_test.py", line 68, in <module>
    fit_least_asymmetry_centering(xgf_imageCube, 15,15,8)
  File "least_asym_test.py", line 51, in fit_least_asymmetry_centering
    half_pix=False, resize=False, weights=False)[0]
  File "/Users/jonathan/anaconda/lib/python3.6/site-packages/least_asymmetry-0.2-py3.6-macosx-10.7-x86_64.egg/least_asymmetry/asym.py", line 248, in actr
    asym = np.array(list(map(make_asym, views, lb_views, w_truth_dup)))
RuntimeError: Arrays must be square

running install
running build
running build_ext
building 'make_asym' extension
gcc -pthread -Wsign-compare -DNDEBUG -g -fwrapv -O3 -Wall -Wstrict-prototypes -fPIC -Iinclude -I/home/jdfraine/anaconda3/include/python3.5m -c make_asym.cc -o build/temp.linux-x86_64-3.5/make_asym.o -std=c++11 -stdlib=libc++ -mmacosx-version-min=10.7
gcc: error: unrecognized command line option ‘-stdlib=libc++’
gcc: error: unrecognized command line option ‘-mmacosx-version-min=10.7’
error: command 'gcc' failed with exit status 1

# if PY_MAJOR_VERSION >= 3
# define NUMPY_IMPORT_ARRAY_RETURN_TYPE int
# else
# define NUMPY_IMPORT_ARRAY_RETURN_TYPE void
# endif

NUMPY_IMPORT_ARRAY_RETURN_TYPE
initmake_asym(void)
{
  Py_InitModule("make_asym", make_asym_methods);
  import_array();
}