PaddleDetection CascadeRCNN 中 num_classes 硬编码 bug
Created by: mzchtx
此处 code 使用了针对 COCO 数据集(num_classes = 81)的硬编码,对于自定义数据预测将出现下述错误:
需要将:
bbox_pred_new = fluid.layers.expand(bbox_pred_new, [1, 81, 1])
修改为:
bbox_pred_new = fluid.layers.expand(bbox_pred_new, [1, self.num_classes, 1])