YOLO-World
YOLO-World copied to clipboard
AILab-CVC
onnx模型导出成功，但是运行推理失败。
感谢非常好的工作。 如题，参考：https://github.com/AILab-CVC/YOLO-World/blob/master/deploy/easydeploy/docs/model_convert.md
生成onnx成功： PYTHONPATH=./ /mnt/bn/tob-lq/miniconda3/envs/yolo-world/bin/python deploy/export_onnx.py configs/finetune_coco_tob/yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco88.py work_dirs/yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco88/epoch_1.pth --custom-text coco_80_cls_name_list.json --opset 11
但是运行 推理报错：
python main_onnxruntime.py
"image_path_to_detect"
/mnt/bn/tob-lq/code/yolo/YOLO-World/work_dirs/epoch_1.onnx
--out-dir work_dir
--img-size 640 640
--show
--score-thr 0.3
--iou-thr 0.7
Traceback (most recent call last):
File "/mnt/bn/tob-lq/code/yolo/YOLO-World/projects/easydeploy/examples/main_onnxruntime.py", line 110, in
发现要加一个type，加完以后新的错误
python main_onnxruntime.py /mnt/bn/tob-lq/code/yolo/YOLO-World/vulgar_multi_cls_20240129_27410.jpg /mnt/bn/tob-lq/code/yolo/YOLO-World/work_dirs/epoch_1.onnx --type yolov8 --out-dir work_dir --img-size 640 640 --show --score-thr 0.3 --iou-thr 0.7
请问如何解决呀，请问有没有具体的例子
我换成 easy_deploy里面的导出代码：
python ./projects/easydeploy/tools/export_onnx.py
configs/finetune_coco_tob/yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco88.py
work_dirs/yolo_world_v2_l_vlpan_bn_sgd_1e-3_40e_8gpus_finetune_coco88/epoch_1.pth
--work-dir work_dir
--img-size 640 640
--batch 1
--device cuda:0
--simplify
--opset 11
--backend 1
--pre-topk 1000
--keep-topk 100
--iou-threshold 0.65
--score-threshold 0.25
x created!
Traceback (most recent call last):
File "/mnt/bn/tob-lq/code/yolo/YOLO-World/./projects/easydeploy/tools/export_onnx.py", line 157, in
