search

yuhuan-wu / P2T

[TPAMI22] Pyramid Pooling Transformer for Scene Understanding
200 stars 18 forks source link

FPS testing #14

Closed LMMMEng closed 1 year ago

LMMMEng commented 1 year ago

Could you please release the code for the fps testing?

yuhuan-wu commented 1 year ago

For the classification, we use the simple code that first runs the model for 100 steps, followed by a torch.cuda.sychronize function, with the batchsize=10.

For detection and segmentation tasks, we directly utilize the test.py of mmsegmentation and mmdetection frameworks.

LMMMEng commented 1 year ago

Many thanks!