Real-time Salient Object Detection

C++ TouchDesigner plugin running U^2-Net at 30 FPS.

Salient Object Detection example animation

I exported U^2-Net as a traced pytorch module and ran it at over 30 FPS inside a custom TouchDesigner plugin with the libtorch CUDA backend. The input resolution is 320x320 RGBA 8-bit, and the output is 320x320 Mono 32-bit float.