search

LikeLy-Journey / SegmenTron

Support PointRend, Fast_SCNN, HRNet, Deeplabv3_plus(xception, resnet, mobilenet), ContextNet, FPENet, DABNet, EdaNet, ENet, Espnetv2, RefineNet, UNet, DANet, HRNet, DFANet, HardNet, LedNet, OCNet, EncNet, DuNet, CGNet, CCNet, BiSeNet, PSPNet, ICNet, FCN, deeplab)
Apache License 2.0
705 stars 162 forks source link

How to test the FPS? #40

Closed y-kl8 closed 4 years ago

y-kl8 commented 4 years ago

image

my test code like this,. The speed of deeplabv3-pluse-mobilenetv2(use ASPP and Decoder) is faster than the deeplabv3-pluse-mobilenetv2(doesn't use ASPP and Decoder). Do somebody know why?

LikeLy-Journey commented 4 years ago

You can reference this: https://github.com/juntang-zhuang/ShelfNet/blob/citys_lw/ShelfNet18_realtime/test_speed.py. Your test script contain the time of copying output tensor from gpu to cpu, which may take much more time than model inference because the size of output tensor is large.