We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 283fd37 commit 40f4ad7Copy full SHA for 40f4ad7
paddle/fluid/inference/tensorrt/convert/yolo_box_op.cc
@@ -50,8 +50,7 @@ class YoloBoxOpConverter : public OpConverter {
50
: 0.5;
51
52
int type_id = static_cast<int>(engine_->WithFp16());
53
- auto input_dim = X_tensor->getDimensions();
54
- auto* yolo_box_plugin = new plugin::YoloBoxPlugin(
+ auto* yolo_box_plugin = new plugin::YoloBoxPluginDynamic(
55
type_id ? nvinfer1::DataType::kHALF : nvinfer1::DataType::kFLOAT,
56
anchors,
57
class_num,
@@ -60,9 +59,7 @@ class YoloBoxOpConverter : public OpConverter {
60
59
clip_bbox,
61
scale_x_y,
62
iou_aware,
63
- iou_aware_factor,
64
- input_dim.d[1],
65
- input_dim.d[2]);
+ iou_aware_factor);
66
67
std::vector<nvinfer1::ITensor*> yolo_box_inputs;
68
yolo_box_inputs.push_back(X_tensor);
0 commit comments