lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.32k stars 264 forks source link

导出模型出错 #253

Open xiaoming-master opened 6 months ago

xiaoming-master commented 6 months ago

当使用export_model.py文件导出模型时, 在PPMSDeformableAttention中, class PPMSDeformableAttention(MSDeformableAttention): ... if reference_points.shape[-1] == 2:

value_spatial_shapes (List): [n_levels, 2], [(H_0, W_0), (H_1, W1), ..., (H{L-1}, W_{L-1})]

        offset_normalizer = paddle.to_tensor(value_spatial_shapes)
        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
        offset_normalizer = offset_normalizer.flip([1]).reshape(
            [1, 1, 1, self.num_levels, 1, 2])

报错:TypeError: Unsupport paddle.assign([Variable, Variable...]) with non-scalar variable. 请问这种情况如何解决呢

lyuwenyu commented 5 months ago

paddle和paddle2onnx版本

xiaoming-master commented 5 months ago

paddle和paddle2onnx版本 paddlepaddle-gpu 2.6.0.post116 paddle2onnx 1.0.6

lyuwenyu commented 5 months ago

建议先用paddlepaddle-gpu==2.4.2试一下

https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_paddle/requirements.txt