「keras(tensorflow)でSegNet」参照に
セグメンテーションを実施しようとしておりますが
予測部分の下記コードが理解できずにおります。
Python
1def predict(test): 2 model = keras.models.load_model('seg_100.h5') 3 probs = model.predict(test, batch_size=1) 4 5 prob = probs[0].reshape((height, width, classes)).argmax(axis=2) 6 return prob 7 8def main(): 9 print("loading data...") 10 ds = dataset.Dataset(test_file='val.txt', classes=classes) 11 test_X, test_y = ds.load_data('test') # need to implement, y shape is (None, 360, 480, classes) 12 test_X = ds.preprocess_inputs(test_X) 13 test_Y = ds.reshape_labels(test_y) 14 15 prob = predict(test_X) 16 writeImage(prob, 'val.png') 17 18if __name__ == '__main__': 19 main()
この中で ds = dataset.Dataset(test_file='val.txt', classes=classes)はval.txt内で
指定されたデータをセットしているように思うのですが、
test_X, test_y = ds.load_data('test')の部分が何を意味しているのか分からずに
おります(特に'test'が何を指しているのか)。
この部分の処理の流れの意味をご教示いただけませんでしょうか。お手数をお掛けいたしますが、
よろしくお願い申し上げます。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2019/01/28 23:35
2019/01/29 08:03
2019/01/29 08:06