學習Faster R-CNN代碼roi_pooling（二）

roi_pooling目錄下 -src文件夾下是c和cuda版本的源碼。 -functions文件夾下的roi_pool.py是繼承了torch.autograd.Function類，實現RoI層的foward和backward函數。class RoIPoolFunction(Function)。 -modules文件夾下的roi_pool.py是繼承了torch.nn.Modules類，實現了對

>>阅读原文<<