前提・実現したいこと
discordbotのプログラムをpythonで書いています
画像が送信された際にその画像を識別するというものを作りたいです
理想としては
python
1@client.event 2async def on_message(message): 3 if message.author.bot: 4 return 5 if message.content.startswith('/pic'): 6 await message.channel.send("少々お待ち下さい") 7 attachment = message.attachments[0] 8 #送信されたファイルをimage.pngという名前で保存する 9 await attachment.save("image.png")
このように/picのコメント付き画像を保存し、kerasとCNNを利用した画像分類のpythonファイルを走らせて、保存したファイルの種類を返信するということをしたいです
「discordbot run」や「discord ファイルを実行する」など色々検索してみましたが見つからず、もし知っている方いましたら教えて下さい。
念の為、画像分類のpythonコードも貼っておきます
python
1# -*- coding: utf-8 -*- 2import numpy as np 3from keras.models import Sequential, model_from_json 4from keras.layers.core import Dense, Dropout 5from keras.optimizers import RMSprop 6from keras.datasets import mnist 7from keras.utils import np_utils, to_categorical 8from keras.preprocessing.image import load_img, img_to_array 9import matplotlib.pyplot as plt 10import cv2 11import os 12 13def main(): 14 # 入力画像のパラメータ 15 img_width = 32 # 入力画像の幅 16 img_height = 32 # 入力画像の高さ 17 img_ch = 3 # 3ch画像(RGB) 18 19 # 入力データ数 20 num_data = 1 21 22 # データ格納用のディレクトリパス 23 SAVE_DATA_DIR_PATH = "E:/python/train/" 24 25 # ラベル 26 labels =["省略"] 27 28 # 保存したモデル構造の読み込み 29 model = model_from_json(open(SAVE_DATA_DIR_PATH + "model.json", 'r').read()) 30 31 # 保存した学習済みの重みを読み込み 32 model.load_weights(SAVE_DATA_DIR_PATH + "weight.hdf5") 33 34 # 画像の読み込み(32×32にリサイズ) 35 # 正規化, 4次元配列に変換(モデルの入力が4次元なので合わせる) 36 img = load_img(SAVE_DATA_DIR_PATH + "test.jpg", target_size=(img_width, img_height)) 37 img = img_to_array(img) 38 img = img.astype('float32')/255.0 39 img = np.array([img]) 40 41 # 分類機に入力データを与えて予測(出力:各クラスの予想確率) 42 y_pred = model.predict(img) 43 44 # 最も確率の高い要素番号 45 number_pred = np.argmax(y_pred) 46 47 # 予測結果の表示 48 print("y_pred:", y_pred) # 出力値 49 print("number_pred:", number_pred) # 最も確率の高い要素番号 50 print('label_pred:', labels[int(number_pred)]) # 予想ラベル(最も確率の高い要素) 51 52 53if __name__ == '__main__': 54 main()
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。