diff --git a/script/cpython/image_functions.py b/script/cpython/image_functions.py index fe0871c..4e12689 100644 --- a/script/cpython/image_functions.py +++ b/script/cpython/image_functions.py @@ -21,5 +21,5 @@ def img_get_int(fname, thres1, thres2, thres3, thres4, header, width, height, d def img_read(fname, header, width, height, depth): img = numpy.fromfile(fname, dtype=numpy.uint32) img.shape = height, width - return img.flat + return img.flatten() \ No newline at end of file diff --git a/script/cpython/wrapper.py b/script/cpython/wrapper.py index b195593..f35b676 100644 --- a/script/cpython/wrapper.py +++ b/script/cpython/wrapper.py @@ -10,7 +10,7 @@ def img_get_int(fname, thres1, thres2, thres3, thres4, header, width, height, de def img_read(fname, header, width, height, depth): ret = call_jep("cpython/image_functions", "img_read", [fname, header, width, height, depth]) - return ret + return ret.data