search

VCIP-RGBD / DFormer

[ICLR 2024] DFormer: Rethinking RGBD Representation Learning for Semantic Segmentation
https://yinbow.github.io/Projects/DFormer/index.html
MIT License
142 stars 24 forks source link

How to check FLOPs #11

Closed catchyoo closed 6 months ago

catchyoo commented 6 months ago

Hello! Thank you for your hard work.

The paper includes calculations of FLOPs in a table, but upon reviewing the code, it's challenging to locate where FLOPs are measured.

Could you possibly tell me where it is measured?"

yinbow commented 6 months ago

Thanks for your attention!

We upload the code for calculations of FLOPs. You can use `PYTHONPATH="$(dirname $0)/..":$PYTHONPATH python benchmark.py --config local_configs.NYUDepthv2.DFormer_Large' to calculate the Flops and Params.

catchyoo commented 6 months ago

Thank you!!