テストセット外の画像に対するラベル付け

https://jp.mathworks.com/help/releases/R2018a/vision/examples/semantic-segmentation-using-deep-learning.html#d119e188

上のURLを参考にセマンティックセグメンテーションを行おうとしているのですが，データセットに元画像をランダムでトリミングしたものを使用しており，学習結果を視覚的にわかりやすくするために元画像上で上のURLの1 つのイメージを使用したネットワーク テストの画像のようにセグメンテーションを行いたいと考えているのですがこれは可能なのでしょうか？