This demo notebook shows image segmentation and removing/adding background with U^2-Net and OpenVINO™.
- Importing Pytorch library and loading U^2-Net model.
- Converting PyTorch U^2-Net model to OpenVINO IR format.
- Loading and preprocessing input image.
- Doing inference on OpenVINO IR model.
- Visualizing results.
@InProceedings{Qin_2020_PR,
title = {U2-Net: Going Deeper with Nested U-Structure for Salient Object Detection},
author = {Qin, Xuebin and Zhang, Zichen and Huang, Chenyang and Dehghan, Masood and Zaiane, Osmar and Jagersand, Martin},
journal = {Pattern Recognition},
volume = {106},
pages = {107404},
year = {2020}
}
If you have not installed all required dependencies, follow the Installation Guide.