File size: 515 Bytes
3f8e08c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# TF for image classification model
import tensorflow
import numpy
from PIL import Image
model = tensorflow.saved_model.load('./')
classes = [ "Safe" , "Naked" , ]
img = Image.open("image.jpg").convert('RGB')
img = img.resize((300, 300 * img.size[1] // img.size[0]), Image.ANTIALIAS)
inp_numpy = numpy.array(img)[None]
inp = tensorflow.constant(inp_numpy, dtype='float32')
class_scores = model(inp)[0].numpy()
print("")
print("class_scores", class_scores)
print("Class : ", classes[class_scores.argmax()]) |