Skip to content

Commit 5740264

Browse files
committed
BUGFIX: save_image was calling incorrect C function
1 parent 5082e00 commit 5740264

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

arrayfire/image.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from .library import *
1111
from .array import *
1212
from .data import constant
13+
import os
1314

1415
def gradient(image):
1516
dx = array()
@@ -18,14 +19,14 @@ def gradient(image):
1819
return dx, dy
1920

2021
def load_image(file_name, is_color=False):
21-
assert(isinstance(file_name, str))
22+
assert(os.path.isfile(file_name))
2223
image = array()
2324
safe_call(clib.af_load_image(pointer(image.arr), c_char_p(file_name.encode('ascii')), is_color))
2425
return image
2526

2627
def save_image(image, file_name):
2728
assert(isinstance(file_name, str))
28-
safe_call(clib.af_load_image(c_char_p(file_name.encode('ascii')), image.arr))
29+
safe_call(clib.af_save_image(c_char_p(file_name.encode('ascii')), image.arr))
2930
return image
3031

3132
def resize(image, scale=None, odim0=None, odim1=None, method=AF_INTERP_NEAREST):

0 commit comments

Comments
 (0)