ResearchPapers/fawkes
ResearchPapers/fawkes
2
0
Fork 0
mirror of https://github.com/Shawn-Shan/fawkes.git synced 2024-12-22 07:09:33 +05:30
Code

Fix Attribute Error with Keras 'img_to_array' in Keras 2.9.0

Browse Source
This commit is contained in:
Longhao.Chen 2022-08-10 01:00:52 +08:00 committed by GitHub
parent aedaa82d22
commit 272469421b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
fawkes/utils.py
View File

@ -35,7 +35,7 @@ import tensorflow as tf
from PIL import Image, ExifTags
from keras.layers import Dense, Activation
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 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):
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.img_to_array(im_data)
return im_data