inference shape is invalid issues;

AILab-CVC / YOLO-World

[CVPR 2024] Real-Time Open-Vocabulary Object Detection

https://www.yoloworld.cc

GNU General Public License v3.0

4.4k stars 426 forks source link

inference shape is invalid issues; #209

Open tutu96177 opened 5 months ago

tutu96177 commented 5 months ago

config: YOLO-World/configs/pretrain/yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_1280ft_lvis_minival.py weights: YOLO-World/scripts_models_export/yolo_world_v2_l_obj365v1_goldg_pretrain_1280ft-9babe3f6.pth inference scripts: YOLO-World/inference.ipynb device : cpu

What causes this problem ?

tutu96177 commented 5 months ago

How to correct being able to run a full infer?

wondervictor commented 5 months ago

Hi @tutu96177, try out the latest version. This bug is temporarily fixed.