Performance issues in libs/models/backbones/efficientnet/utils.py(P2)

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.

https://rotationdetection.readthedocs.io/

Apache License 2.0

1.08k stars 182 forks source link

Performance issues in libs/models/backbones/efficientnet/utils.py(P2) #31

Closed DLPerf closed 3 years ago

DLPerf commented 3 years ago

Hello,I found a performance issue in the definition of build_dataset , libs/models/backbones/efficientnet/utils.py, dataset = dataset.map(_parse_function) was called without num_parallel_calls. I think it will increase the efficiency of your program if you add this.

Here is the documemtation of tensorflow to support this thing.

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.

yangxue0827 commented 3 years ago

Looking forward to your PR！