我刚刚测试了mvtec3d数据集下的bagel在RGB和Fusion2下的训练测试结果一模一样，可以帮我分析一下可能的原因吗谢谢

[rock-lss]

不知道是不是预处理的原因

[TaoTao9]

你好可以请问其他类别也有这样子的情况吗，请问你的测试结果是怎么样的。

------------------ 原始邮件 ------------------ 发件人: "TaoTao9/EasyNet" @.>; 发送时间: 2024年1月26日(星期五) 上午9:24 @.>; @.***>; 主题: [TaoTao9/EasyNet] 我刚刚测试了mvtec3d数据集下的bagel在RGB和Fusion2下的训练测试结果一模一样，可以帮我分析一下可能的原因吗谢谢 (Issue #1)

不知道是不是预处理的原因

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

[TaoTao9]

你好，因为我们提供的权重是Fusion1模式，如果使用Fusion2模式会导致不支持加载（只使用RGB分支），我们会修复这个问题以免引起歧义。

[rock-lss]

但是在fusion1情况下我使用A100显卡会出现显存不够的问题

[rock-lss]

您这边有解决方案吗，我不知道为什么40g的显存不够哈哈哈。简单的说就是bagel在纯RGB和fusion2下面的测试结果一模一样。

[rock-lss]

AUC Image: 0.9741735537190083AP Image: 0.9937716594869666AUC Pixel: 0.7844440231096672AP Pixel: 0.033256266603588595aupro: 0.731035954166466 AUC Image: 0.9741735537190083AP Image: 0.9937716594869666AUC Pixel: 0.7844440231096672AP Pixel: 0.033256266603588595aupro: 0.731035954166466 结果是这样的

[rock-lss]

'bagel': 0.974, 'cable_gland': 1.0, 'carrot': 0.934, 'cookie': 0.951, 'dowel': 0.916, 'foam': 0.934, 'peach' :0.896, 'potato': 0.716, 'rope': 0.992, 'tire': 0.804 平均值：0.912 这是纯RGB下的一组数据

[TaoTao9]

这个我暂时没有想到是什么原因，我这边重新安装和测试都没问题，且显存都是10G以内的占用，我这边机器是A6000，请问可以看一下你的运行命令行吗

[rock-lss]

nohup python trainer.py --gpu_id 0 --obj_id 0 --layer_size 2layer --mode_type Fusion1 > bagel_Fusion1.txt 2>&1 &

[rock-lss]

这样会冲满40g的内存

[rock-lss]

然后我发现在fusion2下训练刚好够用

[rock-lss]

大概fusion2训练需要35GB的显存

[rock-lss]

这是我刚测得的一组数据，在fusion2下训练和测试的结果比纯RGB下低

[TaoTao9]

这是我刚测得的一组数据，在fusion2下训练和测试的结果比纯RGB下低

请你仔细检查你的checkpoint.yaml文件的权重路径，测试结果请使用test.py文件，我们提供的是Fusion1的权重，使用Fusion2模式测试的结果肯定是不正确的。