Add Region Proposal Network for Faster R-CNN, including AnchorLayer, … #2828
Conversation
…RPNLossLayer and ProposalLayer
| initDataLayer( | ||
| configt, &dataLayers, &datas, &layerMap, "anchors", 1, false, use_gpu); | ||
|
|
||
| dataLayers[0]->getOutputValue()->copyFrom(*inputAnchor); |
There was a problem hiding this comment.
It seems that the size of dataLayers[*]->outputvalue is wrong.
You can use LOG(INFO)<< "dataLayers[0] size: "<<dataLayers[0]->getOutputValue()->getElementCnt(); to check if the size of OuptValue equals to (or greater than?) size of inputAnchor.
There was a problem hiding this comment.
Done. I have checked it as you said, and the result showed the same size. This unit test has passed.
|
感谢您给PaddlePaddle贡献代码。由于Paddle V1/V2版本已不再维护，相关代码也已从develop分支上删除，因此关闭您的PR，欢迎您向Paddle最新版-Fluid贡献代码。 |
Add Region Proposal Network for Faster R-CNN, including AnchorLayer, RPNLossLayer and ProposalLayer.
There are correspondences between RPN and SSD: