Tunable U-Net: Controlling image-to-image outputs using a tunable scalar value

S Kang, S Uchida, BK Iwana - IEEE Access, 2021 - ieeexplore.ieee.org
IEEE Access, 2021ieeexplore.ieee.org
Image-to-image conversion tasks are more accurate and sophisticated than ever thanks to
advances in deep learning. However, since typical deep learning models are trained to
perform only one task, multiple trained models are required to perform each task even if they
are related to each other. For example, the popular image-to-image convolutional neural
network, U-Net, is normally trained for a single task. Based on U-Net, this study proposes a
model that outputs variable results using only one trained model. The proposed method …
Image-to-image conversion tasks are more accurate and sophisticated than ever thanks to advances in deep learning. However, since typical deep learning models are trained to perform only one task, multiple trained models are required to perform each task even if they are related to each other. For example, the popular image-to-image convolutional neural network, U-Net, is normally trained for a single task. Based on U-Net, this study proposes a model that outputs variable results using only one trained model. The proposed method produces a continuously changing output by setting an external parameter. We confirm the robustness of our proposed model by evaluating it on binarization and background blurring. According to these evaluations, we confirmed that the proposed model can generate well-predicted outputs by using un-trained tuning parameters as well as the outputs by using trained tuning parameters. Furthermore, the proposed model can generate extrapolated outputs outside the learning range.
ieeexplore.ieee.org
以上显示的是最相近的搜索结果。 查看全部搜索结果