We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 765c70a commit d3a66fcCopy full SHA for d3a66fc
paddle/fluid/operators/detection/generate_proposal_labels_op.cc
@@ -394,6 +394,10 @@ class GenerateProposalLabelsKernel : public framework::OpKernel<T> {
394
auto is_crowd_lod = is_crowd->lod().back();
395
auto gt_boxes_lod = gt_boxes->lod().back();
396
for (int i = 0; i < n; ++i) {
397
+ if (rpn_rois_lod[i] == rpn_rois_lod[i + 1]) {
398
+ lod0.emplace_back(num_rois);
399
+ continue;
400
+ }
401
Tensor rpn_rois_slice =
402
rpn_rois->Slice(rpn_rois_lod[i], rpn_rois_lod[i + 1]);
403
Tensor gt_classes_slice =
0 commit comments