Pipeline hanging after rpn and before performing NMS

zengarden / light_head_rcnn

Light-Head R-CNN

834 stars 222 forks source link

Pipeline hanging after rpn and before performing NMS #68

Closed YellowKyu closed 5 years ago

YellowKyu commented 5 years ago

Hi there,

I performed a quick profiling during inference and found out that the process is hanging right after the RPN and it takes quite a long time, it's 15% of the whole inference !

Any idea on possible reason ?

Thanks !

YellowKyu commented 5 years ago

I switched for TF 1.7 and it solved the issue.