search

AILab-CVC / YOLO-World

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

The setting of category #344

Open XIEFANGYUAN512 opened 3 months ago

XIEFANGYUAN512 commented 3 months ago

I want to detect the attribute of one bounding box, for example, the face, it includes gender, face and mask. How to set the category, i set the category as "male, mask, glasses", is it okay?

wondervictor commented 3 months ago

If you are using the demo, you just input "male, mask, glasses", and it works.

XIEFANGYUAN512 commented 3 months ago

If you are using the demo, you just input "male, mask, glasses", and it works. I mean in the fine-tuning process of my custom dataset, how to set the category of the bounding box. Can it be "male, mask, glasses" ? Also, how are predictions made during inference when the input category does not appear in the vocabulary? Thanks.