Python AIを使って花言葉を判定してみた!

#画像の処理を行うためのライブラリ
from PIL import Image

import numpy as np

#コマンドプロンプトから文字を取り出すためのライブラリ
import sys

import keras

#ニューラルネットワークのモデルを定義するためのライブラリ
from keras.models import Sequential

#畳み込みの処理やプーリングの処理をするためのライブラリ
from keras.layers import Conv2D, MaxPooling2D

#Activationは活性化関数，Dropoutはドロップアウト処理，
#Flattenはデータを1次元に変換，Denseは全結合
from keras.layers import Activation, Dropout, Flatten, Dense

#ファイルをロードするためのライブラリ
from keras.models import load_model

#花言葉集
word1 = "希望"

word2 = "思いやり"

word3 = "幸福"


#クラスを作成
f_classes = ["iris", "grape", "gypsophila"]

f_japanese = ["アヤメ", "ブドウ", "カスミソウ"]


#クラスの数（＝アヤメ，ブドウ，カスミソウの3つ）
num_f_classes = len(f_classes)


#画像のサイズを指定
image_size = 50

#ニューラルネットワークのモデルを作成する
def neural_model():
        
    model = Sequential()
    #------------------------------------------------
    model.add(Conv2D(32, 
                     kernel_size=(3,3), 
                     padding="same",
                     input_shape=(50, 50, 3)))
    
    #マイナスの値を除去
    model.add(Activation("relu"))
    #------------------------------------------------ 
    model.add(Conv2D(32, 
                     kernel_size=(3,3)))
    
    model.add(Activation("relu"))
    
    model.add(MaxPooling2D(pool_size=(2,2))) 
    
    #過学習の抑制(ランダムでニューロンを無効化)
    #偏ったニューロンに依存しないため
    model.add(Dropout(rate=0.25))#25%無効化
    
    
    #------------------------------------------------
    model.add(Conv2D(64, 
                     kernel_size=(3,3), 
                     padding="same"))
    
    model.add(Activation("relu"))
    #------------------------------------------------
    model.add(Conv2D(64, 
                     kernel_size=(3,3)))
    
    model.add(Activation("relu"))
    
    model.add(MaxPooling2D(pool_size=(2,2)))
    
    model.add(Dropout(0.25))
    
    
    #------------------------------------------------
    model.add(Flatten())
    
    model.add(Dense(512))
    
    model.add(Activation("relu"))
    
    model.add(Dropout(0.5))
    
    model.add(Dense(num_f_classes))
    
    model.add(Activation("softmax"))
    
    
    #------------------------------------------------
    #オプティマイザーを作成
    opt = keras.optimizers.RMSprop(learning_rate=0.0001, 
                                   decay=1e-6)
    
    #コンパイル
    model.compile(loss="categorical_crossentropy", 
                  optimizer=opt, 
                  metrics=["accuracy"])
    
    #モデルのロード（）
    model = load_model("./flower_cnn.h5")
    
    return model

def main():
    #コマンドプロンプトの2番目の引数
    image = Image.open(sys.argv[1])
    
    #画像を表示
    show_image = image.show(image)
    
    #RGB値に変換する
    image = image.convert("RGB")
    #リサイズ
    image = image.resize((image_size, image_size))
    
    #画像を数値に変換する
    data = np.asarray(image)
    
    #リストを用意
    X = []
    
    #Xにdataを入れる
    X.append(data)
    
    #XをTensorFlowが扱いやすいようにNumpy型に変換する
    X = np.array(X)
    
    #ニューラルネットワークのモデル関数を呼び出す
    model = neural_model()
    
    #画像をニューラルネットワークのモデルから予測する
    result = model.predict([X])[0]
    
    #一番値の大きい配列の添え字を返す
    pred = result.argmax()
    
    #精度を求める(?)
    percentage = int(result[pred]*100)
    
    #結果を表示
    print("\n{0} {1} %".format(f_classes[pred], percentage))
    
    
    #花言葉を表示
    if f_classes[pred] == "iris":
        
        print("\n{0}の花言葉は『{1}』です．".format(f_japanese[pred], word1))
        
    elif f_classes[pred] == "grape":
        
        print("\n{0}の花言葉は『{1}』です．".format(f_japanese[pred], word2))
    
    elif f_classes[pred] == "gypsophila":
        
        print("\n{0}の花言葉は『{1}』です．".format(f_japanese[pred], word3))
          

# コマンドプロンプト上で実行  
if __name__ == "__main__":
    main()