Fix Image.frombytes for older version of Python/PIL

This commit is contained in:
Stefan van der Walt
2014-02-12 12:59:51 +02:00
parent c930133982
commit b71008a135
+7 -1
View File
@@ -109,7 +109,13 @@ def imsave(fname, arr, format_str=None):
if not isinstance(fname, string_types) and format_str is None:
format_str = "PNG"
img = Image.frombytes(mode, (arr.shape[1], arr.shape[0]), arr.tostring())
try:
img = Image.frombytes(mode, (arr.shape[1], arr.shape[0]),
arr.tostring())
except AttributeError:
img = Image.fromstring(mode, (arr.shape[1], arr.shape[0]),
arr.tostring())
img.save(fname, format=format_str)