RuiyingLu / HVQ-Trans

Code for "Hierarchical Vector Quantized Transformer for Multi-class Unsupervised Anomaly Detection" published in NeurIPS 2023
26 stars 2 forks source link

Codebook 的选择 #2

Open hlr7999 opened 6 months ago

hlr7999 commented 6 months ago

您好,您论文中 codebook 是通过训练一个分类器来选择的,但我看代码中好像是直接给的 label 选择的?

RuiyingLu commented 6 months ago

您好!感谢关注。由于工业检测数据集各类别之间区别度很高,分类器训练难度很低,精度接近100%。开学后我们可以发您一版带分类器训练的代码。

---- 回复的原邮件 ---- | 发件人 | Liren @.> | | 发送日期 | 2024年02月14日 21:05 | | 收件人 | RuiyingLu/HVQ-Trans @.> | | 抄送人 | Subscribed @.***> | | 主题 | [RuiyingLu/HVQ-Trans] Codebook 的选择 (Issue #2) |

您好,您论文中 codebook 是通过训练一个分类器来选择的,但我看代码中好像是直接给的 label 选择的?

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.***>

leolin65 commented 1 month ago

你好,

可以也给我一份带分类器训练的代码,非常感谢您。