faster-rcnn-pytorch icon indicating copy to clipboard operation
faster-rcnn-pytorch copied to clipboard
verified publisher icon bubbliiiing

rewrite rpn operation(from numpy to pytorch)

Open n719710930 opened this issue 3 years ago • 2 comments

我发现frcnn_trainning.py中的numpy操作极大地影响了训练速度，而且会导致CPU利用率太高，所以我将其中绝大部分numpy操作改为了pytorch操作，在我的实验设置下提速10%左右，每个迭代由1.33 it/s 降低为 1.13it/s，任保留了作者之前的numpy写法，是可选择的。

n719710930 avatar Jul 01 '22 07:07 n719710930

我周末找个时间研究一哈。

bubbliiiing avatar Jul 01 '22 14:07 bubbliiiing

Hello，我发现它在低版本的pytorch中会出现bug，我抓耳挠腮了😂

bubbliiiing avatar Jul 04 '22 14:07 bubbliiiing