Рет қаралды 10
оригинал - github.com/AILab-CVC/YOLO-World
python image_demo.py configs/pretrain/yolo_world_l_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py yolow-v8_l_clipv2_frozen_t2iv2_bn_o365_goldg_pretrain.pth zoom_30.jpg 'bicycle' --topk 100 --threshold 0.005 --output-dir demo_outputs
для выделения других объектов поменять промт - 'bicycle'
скачать и переименовать веса и конфиг:
wget huggingface.co/spaces/steveng...
mv yolow-v8_l_clipv2_frozen_t2iv2_bn_o365_goldg_pretrain.pth?download=true yolow-v8_l_clipv2_frozen_t2iv2_bn_o365_goldg_pretrain.pth
wget huggingface.co/spaces/steveng...
mv yolo_world_l_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py?download=true yolo_world_l_t2i_bn_2e-4_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py
при установке с cuda torch могут быть проблемы с mmcv, которую пришлось собирать отдельно.