search

AILab-CVC / YOLO-World

[CVPR 2024] Real-Time Open-Vocabulary Object Detection
https://www.yoloworld.cc
GNU General Public License v3.0
4.64k stars 449 forks source link

export onnx model issues. #217

Open tutu96177 opened 7 months ago

tutu96177 commented 7 months ago

I want to export an onnx model with clip and yolov8 backbone; configs: YOLO-World/configs/pretrain/yolo_world_v2_l_clip_large_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_800ft_lvis_minival.py YOLO-World/yolo_world_v2_l_clip_large_o365v1_goldg_pretrain_800ft-9df82e55.pth The command is as follows: python deploy/export_onnx.py --opset 11 --simplify --model-only The following problems occur when exporting: 图片

I changed two models to export, but only the model of yolov8 backbone is exported; The changes are as follows: 图片 图片

tutu96177 commented 7 months ago

In addition, how to export a model with clip and yolov8 backbone, so far, the export is yolov8 backbone's onnx model?