ptyhonではじめて画像分類のプログラムを以下のサイトを参考に書きました。
<Deep Learningで犬・猫を分類してみよう-ai_academy-https://aiacademy.jp/texts/show/?id=164>
認識できるクラスが無ければ新たにクラスを追加し、もう一度学習を行うことができるように変えてみました。
しかし、この場合もう一度学習を行う際も一度目の学習と同じようにかなりの時間がかかります。追加したクラスだけを学習させ、学習時間の短縮を行うことは可能でしょうか。ご教授いただければ幸いです。
python
1from h5py._hl.files import File 2import numpy as np 3from keras.models import load_model 4from PIL import Image 5from matplotlib import pyplot as plt 6import pickle 7from icrawler.builtin import BingImageCrawler 8import shutil 9 10imsize = (64, 64) 11with open('classes',mode='rb')as f: 12 classes=pickle.load(f) 13 14 15while True: 16 imagess=input('画像を入力してください>>') 17 testpic= imagess 18 keras_param = "./cnn.h5" 19 20 def load_image(path): 21 img = Image.open(path) 22 img = img.convert('RGB') 23 24 img = img.resize(imsize) 25 # 画像データをnumpy配列の形式に変更 26 img = np.asarray(img) 27 img = img / 255.0 28 return img 29 30 model = load_model(keras_param) 31 img = load_image(testpic) 32 prd = model.predict(np.array([img])) 33 print(prd) # 精度の表示 34 prelabel = np.argmax(prd, axis=1) 35 36 print(classes) 37 answer=input('この中に該当のクラスはありますかy/n>>') 38 if answer== "y" : 39 for f in range(len(classes)): 40 if prelabel == f: 41 print(classes[f]) 42 answer2=input("正解ですかy/n>>") 43 if answer2=="y": 44 print("ok") 45 elif answer2=="n": 46 print(classes) 47 k=input('該当のクラスを指定してください>>') 48 shutil.copy(imagess, './images/'+k) 49 elif answer=="n": 50 name = input("新たなクラスを入力して下さい>>") 51 crawler = BingImageCrawler(storage={"root_dir": name}) 52 crawler.crawl(keyword=name, max_num=100) 53 new_path = shutil.move(name, 'images/'+name) 54 import dateset 55 import train 56
Layer
1Model: "sequential" 2_________________________________________________________________ 3 Layer (type) Output Shape Param # 4================================================================= 5 conv2d (Conv2D) (None, 64, 64, 32) 896 6 7 activation (Activation) (None, 64, 64, 32) 0 8 9 conv2d_1 (Conv2D) (None, 62, 62, 32) 9248 10 11 activation_1 (Activation) (None, 62, 62, 32) 0 12 13 max_pooling2d (MaxPooling2D (None, 31, 31, 32) 0 14 ) 15 16 dropout (Dropout) (None, 31, 31, 32) 0 17 18 conv2d_2 (Conv2D) (None, 31, 31, 64) 18496 19 20 activation_2 (Activation) (None, 31, 31, 64) 0 21 22 conv2d_3 (Conv2D) (None, 29, 29, 64) 36928 23 24 activation_3 (Activation) (None, 29, 29, 64) 0 25 26 max_pooling2d_1 (MaxPooling (None, 14, 14, 64) 0 27 2D) 28 29 dropout_1 (Dropout) (None, 14, 14, 64) 0 30 31 dense (Dense) (None, 512) 6423040 32 33 activation_4 (Activation) (None, 512) 0 34 35 dropout_2 (Dropout) (None, 512) 0 36 37 dense_1 (Dense) (None, 3) 1539 38 39 activation_5 (Activation) (None, 3) 0 40 41================================================================= 42Total params: 6,490,147 43Trainable params: 6,490,147 44Non-trainable params: 0
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。