TaoTao9 / EasyNet

11 stars 0 forks source link

在fusion2模式下训练,在fusion1模式下测试是可行的吗,我这样做好像数据更差了 #2

Closed rock-lss closed 9 months ago

TaoTao9 commented 9 months ago

请按照对应模式进行测试,Fusion1显存过大请适当调节batch大小,也即是'--bs'参数,可以设置小点以满足机器的显存。

rock-lss commented 9 months ago

好的,谢谢

rock-lss commented 9 months ago

屏幕截图 2024-02-01 141804 这是我重新在fusion1下训练测试后得到的数据,不知道为什么还是不如纯RGB下的

TaoTao9 commented 9 months ago

屏幕截图 2024-02-01 141804 这是我重新在fusion1下训练测试后得到的数据,不知道为什么还是不如纯RGB下的

因为重建任务有多个损失收敛,训练过程波动比较大,训练到能收敛比较困难,我们在训练中也做了多次调节,重建任务缺失会比较难训练。

TaoTao9 commented 9 months ago

请问你做的是3D AD这方面的研究工作吗

rock-lss commented 9 months ago

我是东南大学的研究生,目前在贵阳航天所从事元器件表面缺陷检测方面的研究

TaoTao9 commented 9 months ago

这方面可以关注下https://github.com/M-3LAB/awesome-industrial-anomaly-detection 的survey,目前这方面的工作M3DM和CPMF已经做到SOTA。

rock-lss commented 9 months ago

谢谢