6. 推論

6.1 推論

前章までで作成されたモデル（ニューラルネットワークと最適化されたパラメータの組み合わせ）を用いて、 地中レーダー画像の測定値または何らかの方法で作成されたものから誘電率分布を推定することを推論(inference)と呼びます。

図6-1に推論の手順を示します。
以下の2つのファイルを用意します。

1. fdtd.bin: FDTD法で出力されるfdtd.binと同じ書式で、 地中レーダー画像を代入したもの。 誘電率は不要なのでダミー（例えばすべて1）を代入する。
2. dlgpr.pth: 前章までで作成されたモデルファイル

推論を行うにはファイルdlgpr.pyを以下のように編集してください。

リスト6-1 推論の設定（dlgpr.pyの一部）

    # modelfile
    modelfile = 'dlgpr.pth'   # modelfile名
    load_model = 1     # modelfileを読み込むので1とする
    save_model = 0     # modelfileを保存しないので0とする

    # 計算パラメーター
    ndata = 1          # データ数（1以上、通常1)
    batch_size = 50    # バッチサイズ,  結果に関係ないのでダミー(50のままでよい）
    num_epochs = 0     # エポック数, 反復計算をしないので0とする
    train_ratio = 0    # 訓練しないので0とする

推論を行うには下記のコマンドを実行してください。計算時間は一瞬です。

$ python dlgpr.py

推論結果を図形出力するにはplot.pyを下記のように編集した後、 下記のコマンドを実行してください。

_plot2d_labels((6, 4), 1, 1, labels, preds, dt)

$ python plot.py

図6-2に2次元モデルの図形出力例を示します。右が推論結果です。
3次元モデルのときは図4-8のような図が出力されます。