keras python このコードの意味

Question

現在　pythonとkeras を使い　ある書籍の内容にある、kaggle の　猫と犬の分類の問題をといていて下記のコードが出てきました。

しかし、下記コードの
features[i * batch_size:( i + 1 )* batch_size] = features_batch
labels[i * batch_size:(i + 1) * batch_size] = labels_batch

の部分が具体的にどんなことをしてるのかがいまいちわかりません。。。
features[20:21]　に　features_batch をいれている？
でもなぜ、20：21なの？
って感じです。アドバイスお願いします。

base_dir = "./cat_and_dogs_small"

train_dir = os.path.join(base_dir,"train")
validation_dir = os.path.join(base_dir,"validation")
test_dir = os.path.join(base_dir,"test")

datagen = ImageDataGenerator(rescale=1./255)
batch_size=20


def extract_features(directory,sample_count):
    #　0の配列を作成する。
    features = np.zeros(shape=(sample_count,4,4,512))
    labels = np.zeros(shape=(sample_count))
    generator = datagen.flow_from_directory(directory,
                                      target_size=(150,150),
                                      batch_size=batch_size,
                                      class_mode="binary")
    
    i = 0
    for inputs_batch,labels_batch in generator:
        
        features_batch = conv_base.predict(inputs_batch)
        features[i * batch_size:( i + 1 )* batch_size] = features_batch
        labels[i * batch_size:(i + 1) * batch_size] = labels_batch
        i += 1
    
        if i * batch_size >= sample_count:
            break
        
    return features,labels

train_features,train_labels = extract_features(train_dir,2000)
validation_features,validation_labels = extract_features(validation_dir,1000)
test_features,test_labels = extract_features(test_dir,1000)
コード

Comments

[quickquip]

どういう手法を使っているか、手法の名前など書いてないですか? （アンサンブルとか転移学習とか、そういう名前です） 「下記のコードが出てきました」というのはどこに出てきたのでしょうか? その書籍ですか?

Answer 1

こんにちは。

features[i * batch_size:( i + 1 )* batch_size] = features_batch

上のコードでbatch_size = 20だったとき
features[0:20], features[20:40], features[40:60], と増えていくのはわかるでしょうか。

ここでは特徴（feature）と正解（label）のリストに
指定フォルダから読み込んだデータをバッチサイズずつ詰め込んでいます。

Comments

[nobita]

なるほど！！！
ありがとうございます。
すっきりしました！！