Skip to content

RT-DETR-L-wireless_table_cell_det cannot export ONNX #16593

@naonao-beibei

Description

@naonao-beibei

🔎 Search before asking

  • I have searched the PaddleOCR Docs and found no similar bug report.
  • I have searched the PaddleOCR Issues and found no similar bug report.
  • I have searched the PaddleOCR Discussions and found no similar bug report.

🐛 Bug (问题描述)

Unable to find the complete yml configuration file, resulting in the inability to convert from training model pdparams to inference model;
And the pdmodel file for RT-DETR-L-wireless_table_cell_det cannot be found from the download link of the thrust model, resulting in the inability to convert to ONNX.

🏃‍♂️ Environment (运行环境)

🌰 Minimal Reproducible Example (最小可复现问题的Demo)

无法找到完整的配置文件`Global:
model: RT-DETR-L_wireless_table_cell_det
mode: check_dataset # check_dataset/train/evaluate/predict
dataset_dir: "/paddle/dataset/paddlex/cells_det/cells_det_coco_examples"
device: gpu:0,1,2,3
output: "output"

CheckDataset:
convert:
enable: False
src_dataset_type: null
split:
enable: False
train_percent: null
val_percent: null

Train:
num_classes: 1
epochs_iters: 40
batch_size: 8
learning_rate: 0.0001
pretrain_weight_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/RT-DETR-L_wireless_table_cell_det_pretrained.pdparams"
warmup_steps: 100
resume_path: null
log_interval: 10
eval_interval: 1

Evaluate:
weight_path: "output/best_model/best_model.pdparams"
log_interval: 10

Export:
weight_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/RT-DETR-L_wireless_table_cell_det_pretrained.pdparams"

Predict:
batch_size: 1
model_dir: "output/best_model/inference"
input: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/table_recognition.jpg"
kernel_option:
run_mode: paddle
`

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions