技术栈
开放检测
早茶和猫
6 天前
yolo
·
目标检测
·
prompt
·
yoloe
·
视觉提示
·
开放检测
【YOLOE: Real-Time Seeing Anything】predict_visual_prompt.py视觉推理代码分析（检测版本）
一、YOLODataset中build_transforms加载LoadVisualPromptLoadVisualPrompt位于augment.py用于读取视觉提示特征。将标注gt的box或者mask转成特征图大小的mask，存储在labels[“visuals”]中