mirror of
https://github.com/Shawn-Shan/fawkes.git
synced 2024-12-22 07:09:33 +05:30
Fix Attribute Error with Keras 'img_to_array' in Keras 2.9.0
This commit is contained in:
parent
aedaa82d22
commit
272469421b
@ -35,7 +35,7 @@ import tensorflow as tf
|
|||||||
from PIL import Image, ExifTags
|
from PIL import Image, ExifTags
|
||||||
from keras.layers import Dense, Activation
|
from keras.layers import Dense, Activation
|
||||||
from keras.models import Model
|
from keras.models import Model
|
||||||
from keras.preprocessing import image
|
from keras.utils import image_utils as image
|
||||||
|
|
||||||
from fawkes.align_face import align
|
from fawkes.align_face import align
|
||||||
from six.moves.urllib.request import urlopen
|
from six.moves.urllib.request import urlopen
|
||||||
@ -263,7 +263,7 @@ def load_victim_model(number_classes, teacher_model=None, end2end=False):
|
|||||||
|
|
||||||
def resize(img, sz):
|
def resize(img, sz):
|
||||||
assert np.min(img) >= 0 and np.max(img) <= 255.0
|
assert np.min(img) >= 0 and np.max(img) <= 255.0
|
||||||
from keras.preprocessing import image
|
from keras.utils import image_utils as image
|
||||||
im_data = image.array_to_img(img).resize((sz[1], sz[0]))
|
im_data = image.array_to_img(img).resize((sz[1], sz[0]))
|
||||||
im_data = image.img_to_array(im_data)
|
im_data = image.img_to_array(im_data)
|
||||||
return im_data
|
return im_data
|
||||||
|
Loading…
Reference in New Issue
Block a user