search

wbenbihi / hourglasstensorflow

Tensorflow implementation of Stacked Hourglass Networks for Human Pose Estimation
MIT License
479 stars 177 forks source link

Performance issue in the definition of _attention_part_crf, hourglass_tiny.py(P1) #70

Closed DLPerf closed 2 years ago

DLPerf commented 3 years ago

Hello, I found a performance issue in the definition of _attention_part_crf, hourglass_tiny.py, tf.pad will be created repeatedly during program execution, resulting in reduced efficiency. I think both att and pad should be created before the loop.

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

DLPerf commented 3 years ago

@wbenbihi Hi, my friend, could you consider my issue asap?