Keras: CNN画像分類 (クラスの可視化)

2018年7月10日 Keras 2.2 TensorFlow 1.8 #Keras #TensorFlow #VGG16

プログラミング

# -*- encoding: utf-8 -*-
from keras.applications.vgg16 import VGG16, preprocess_input, decode_predictions
from keras.preprocessing import image
import numpy as np

def criterion(file):

  img = image.load_img(file, target_size=(224, 224))
  img = image.img_to_array(img)
  img = np.expand_dims(img, axis=0)
  img = preprocess_input(img)

  model = VGG16(weights="imagenet")

  probs = model.predict(img)
  print("PREDICT: ", decode_predictions(probs, top=3)[0])

  print(np.argmax(probs[0]))

if __name__ == "__main__":
  import sys, keras, tensorflow
  print("Python %s" % sys.version)
  print("Keras %s" % keras.__version__)
  print("TensorFlow %s" % tensorflow.__version__)

  criterion(sys.argv[1])