search

JiehongLin / SAM-6D

[CVPR2024] Code for "SAM-6D: Segment Anything Model Meets Zero-Shot 6D Object Pose Estimation".
366 stars 30 forks source link

Tried using some modules from sam6d but got poor results after setting model.eval() during testing #55

Open W-QY opened 6 months ago

W-QY commented 6 months ago

First of all thank you very much for your work!

I tried to add the "Fine Point Matching" module in your code to my network. During the training process, the loss converged very well, but the test results were poor. After I removed "model.eval()", the model test results were very good. I would like to ask you if you know the reason?

huassaue commented 3 weeks ago

是不是batch_size设置过小