FuNz-0 / PromptAD

Official implementation of CVPR 2024 PromptAD: Learning Prompts with Only Normal Samples for Few-Shot Anomaly Detection
The Unlicense
78 stars 4 forks source link

train_cls.py的epoch #17

Open Nianwenxin opened 2 weeks ago

Nianwenxin commented 2 weeks ago

请问作者的实验中,train_cls.py的epoch设置为多少?因为train_cls.py中的epoch默认是1,直接运行的结果和论文中差很多。考虑到有可能是和n_ctx_ab写反了,epoch改成100后,虽然roc值有提升,可是和论文里还是有写差距。比如在MVTec数据集上2-shot情况,我跑出来的结果是93.48,而论文中对应的结果是95.7上下1.5(94.2~97.2)。

FuNz-0 commented 1 week ago

您好,我昨天用上述参数分别在windows和ubuntu两个系统下对2-shot的结果进行了复现,都能实现94.2~97.2的结果。我们之前的实验就发现在不同的机器下few-shot的结果会有很大差异,而且采样对结果的影响也很大,所以我们把采样给固定了,你可以试试随机采样的效果。

@.***

发件人: Nianwenxin 发送时间: 2024-09-01 22:03 收件人: FuNz-0/PromptAD 抄送: Subscribed 主题: [FuNz-0/PromptAD] train_cls.py的epoch (Issue #17) 请问作者的实验中,train_cls.py的epoch设置为多少?因为train_cls.py中的epoch默认是1,直接运行的结果和论文中差很多。考虑到有可能是和n_ctx_ab写反了,epoch改成100后,虽然roc值有提升,可是和论文里还是有写差距。比如在MVTec数据集上2-shot情况,我跑出来的结果是93.48,而论文中对应的结果是95.7上下1.5(94.2~97.2)。 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.***>

fftao294 commented 1 week ago

我也遇到了同样的问题,请问一下,在哪里修改使用随机采样?

FuNz-0 commented 1 week ago

数据集加载部分,可以修改

@.***

发件人: fftao294 发送时间: 2024-09-06 14:05 收件人: FuNz-0/PromptAD 抄送: FuNz; Comment 主题: Re: [FuNz-0/PromptAD] train_cls.py的epoch (Issue #17) 我也遇到了同样的问题,请问一下,在哪里修改使用随机采样? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>