RT-DETR训练好的pth导出为onnx模型报错记录

export_onnx.py文件

if __name__ == '__main__':

    parser = argparse.ArgumentParser()
    parser.add_argument('--config', '-c', type=str, default="../configs/rtdetr/rtdetr_r50vd_6x_coco.yml" )
    parser.add_argument('--resume', '-r', type=str, default="./output/rtdetr_r50vd_6x_coco/checkpoint0095.pth")
    parser.add_argument('--file-name', '-f', type=str, default='detection_model.onnx')
    parser.add_argument('--check',  action='store_true', default=False,)
    parser.add_argument('--simplify',  action='store_true', default=False,)

    args = parser.parse_args()

    main(args)

修改了--config和--resume的default内容。

运行：python export_onnx.py出现报错

TypeError:bases must be types

这个时候可能需要重新安装一下：pip install protobuf==3.20.2

再次执行：python export_onnx.py 

获得导出模型：