2020-10-24から1日間の記事一覧

【U-Net】PytorchでSemantic Segmentationの推論+切り抜きを実装

前書き melheaven.hatenadiary.jp前回、上記の記事のように学習を進めました。 今回は推論をやっていきます。 推論の準備 学習後、作成した重みファイル(.pth)をロードします。 # 学習モデルを引っ張ってくる(前回の記事参照) model = ResNetUNet(num_clas…