U²-Net Portrait on UE4

Example UE4 project for U²-Net Portrait.

u2net-demo.mp4

(Note that environment assets in this video is not included. You can get the asset from UE market place for free)

Environment

• OS: Windows 10 64bit
• Unreal Engine: 4.26.2
• NNEngine plugin v1.2 or above

Download

Please download from the release page.
(Or, clone this repository, download ONNX model from here (GitHub release page) or here (Google Drive), and place it under Source/ThirdParty/Models.)

Run the demo

1. Extract the downloaded zip file and double-click U2NetPortrait.uproject.
2. After launching, follow the tutorial of /Content/NNEngineDemo-U2NetPortrait/Tutorial.uasset.

Model details

See the following pages for the details of the model used in this project.

• U²-Net: Going Deeper with Nested U-Structure for Salient Object Detection
• GitHub