search

xjgaocs / Trans-SVNet

26 stars 5 forks source link

Trans_SVNet结果无法复现 #1

Open aunusualman opened 3 years ago

aunusualman commented 3 years ago

您好，我在第二步train tecno达到和您类似的效果，但是train_sv效果却很差，也是按照您的的参数，能否提供训练好的模型参数呢

xjgaocs commented 3 years ago

Transformer 不是end-to-end trained，input embedding是fixed，而且一两个epoch就收敛

aunusualman commented 3 years ago

是的，我的实验结果也是类似，但是效果比论文中差很多，请问您提取帧率是1HZ还是5hz呢? 我是按照1HZ来做的，input embedding也是参考您的代码实现的。

xjgaocs commented 3 years ago

1Hz training，在5Hz test也没问题。有了很好的embedding之后，Transformer自带attention机制，不应该降低performance。

aunusualman commented 3 years ago

您好，感谢您的指点。您的意思是说训练用的1HZ，test用的5hz吗？ 我利用Tecno得到的embedding结果也和您代码中记录的val_8924_test_8603差不多，但是第三阶段就效果比较差了，参数也是用的默认的，您觉得可能是什么原因呢？

xjgaocs commented 3 years ago

1Hz training，在1Hz和5Hz test都没问题。我上传了训练好的模型参数。

aunusualman commented 3 years ago

非常感谢！

aunusualman commented 3 years ago

Hi, 抱歉再次打扰您，我在1hz下用res50提取特征阶段，precision和recall 只有0.73和0.748左右，低于您论文中的0.78结果，我怀疑是我模型效果差的主要原因。我参考的是您训练embedding的代码，图片预处理参考TMRNet，做了黑边剪切处理，请问您是否还做了其他的一些处理呢？ 非常感谢您的帮助！