6DRepNet360 icon indicating copy to clipboard operation
6DRepNet360 copied to clipboard

Fine-tuned model?

Open Redhwan-A opened this issue 10 months ago • 4 comments

Hi,

Could you make your fine-tuned model available, as you did in 6DRepNet? please.

I would like to test my own datasets on your own fine-tuned model for the comparison for narrow and full-range datasets.

Thank you in advance!

Redhwan-A avatar Apr 11 '24 07:04 Redhwan-A

Hello,

you will find a link to the main model here: https://github.com/thohemp/6DRepNet360/blob/master/sixdrepnet360/test.py#L140 The other ones will be added soon.

thohemp avatar Apr 11 '24 08:04 thohemp

@thohemp, thank you so much for your reply. This fine-tuned model for narrow range ("https://cloud.ovgu.de/s/2sP3yLrEwyfSmqC/download/6DRepNet360_300W_LP_AFLW2000.pth").

Really, I am interested in the fine-tuned model for the full range for CMU datasets.

Thanks.

Redhwan-A avatar Apr 11 '24 10:04 Redhwan-A

I think the notation of the file is misleading. I updated the readme with both models.

thohemp avatar Apr 11 '24 10:04 thohemp

Thank you so much for your update. I ran your code on the CMU dataset and used your fine-tuned model. I got this result for CMU dataset for my CMU dataset:

Yaw: 7.0174, Pitch: 10.5485, Roll: 7.6579, MAE: 8.4079
Vec1: 11.5907, Vec2: 8.3935, Vec3: 11.6801, VMAE: 10.5548

360

Redhwan-A avatar Apr 12 '24 10:04 Redhwan-A