search

xuebinqin / U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
Apache License 2.0
8.67k stars 1.5k forks source link

Inference time is slow #282

Open dbAIStudio opened 2 years ago

dbAIStudio commented 2 years ago

u2net_test.py It took more than 1 second to detect a single image of 2080, but you can see that it is 30 frames in the 1080 test, what is the reason for this?

rdutta1999 commented 2 years ago

There could be multiple reasons for slow inference: 1) Slow CPU / GPU. 2) Not enough RAM leading to paging. 3) More parameters have been added (probably by changing the output shape), thus taking more computation time.