Question about learnable parameters.

SysCV / sam-hq

Segment Anything in High Quality [NeurIPS 2023]

https://arxiv.org/abs/2306.01567

Apache License 2.0

3.73k stars 224 forks source link

Closed Nastu-Ho closed 1 year ago

Nastu-Ho commented 1 year ago

The learnable parameters calculated according to the program are about 1.3M. But the paper says 5.1M, and I'm not quite sure what that's going on.

lkeab commented 1 year ago

Hi, please refer to the discussion here. The unit is different.

Nastu-Ho commented 1 year ago

Could you teach me how to divide SAM's raw weights into image_encoder, prompt_encoder, mask_decoder to store them?