whai362 / PSENet

Official Pytorch implementations of PSENet.
Apache License 2.0
1.17k stars 344 forks source link

fps #45

Open wangshuai66666 opened 5 years ago

wangshuai66666 commented 5 years ago

运行作者的test_ic15.py,fps只有0.65,我的GPU是k40c 12G的。请问怎么提高速度呢?

Annihilation7 commented 5 years ago

我更慢。。在1s模式下,一张图40秒左右。。但是没用c++版本的pse算法,用的python的。不过这也太慢了吧- -

wangshuai66666 commented 5 years ago

我更慢。。在1s模式下,一张图40秒左右。。但是没用c++版本的pse算法,用的python的。不过这也太慢了吧- -

我用的是c++版本的pse,不知道速度慢是哪里的问题。

saicoco commented 5 years ago

如果图缩放1/4, fps可以到20+

wangshuai66666 commented 5 years ago

如果图缩放1/4, fps可以到20+ 可是这样的话,正确率就降低很多了。

whai362 commented 5 years ago

我测试是用1080ti测试的 提高速度可以在执行测试脚本的时候用"--scale 4"

wangshuai66666 commented 5 years ago

我测试是用1080ti测试的 提高速度可以在执行测试脚本的时候用"--scale 4"

我试了,输出图片是小了很多,但是fps还只是0.6。我的gpu是k40c,性能也不错吧。 @whai362

xianchen22 commented 5 years ago

还可以从哪些方面提高速度?

wangshuai66666 commented 5 years ago

我发现 把 --long_size设置小一些,速度会快。但是改变--scale 速度没有变化,只是输出图像小了,这是为什么呢?我现在--long_size设置1024,基本可以保证正确率,fps只有2.3。 @whai362 @saicoco