whai362 / pan_pp.pytorch

Official implementations of PSENet, PAN and PAN++.
Apache License 2.0
435 stars 90 forks source link

PA function is much slower than pse function during my imgs #81

Open Zhang-O opened 2 years ago

Zhang-O commented 2 years ago

test on a test dataset of small imgs (about 600 pics),panet takes more time than psenet, average 20ms higher than psenet test on a test dataset of big imgs (about 30 pics),panet takes more time than psenet, average 100ms higher than psenet

Than I try to figture it out and find the obivious difference between the two model is the pa module which takes more time than pse module with input of same size. I have no idea how could this happened?

Zhang-O commented 2 years ago

Further test, when img size is (h,w,c)=(1280, 1280, 3), the pa process(the input size of pa is 1/4 of the input of resnet-18 ) takes about 100 ms with about 10 text lines, while the other parts of the network before the pa takes about 10ms.