search

syp2ysy / VRP-SAM

[CVPR 2024] Official implementation of "VRP-SAM: SAM with Visual Reference Prompt"
MIT License
74 stars 8 forks source link

When VRP encoder is DINOv2, how to define mid-level and high-level features? #10

Open suyan451 opened 2 months ago

suyan451 commented 2 months ago

When VRP encoder is DINOv2, how to define mid-level and high-level features?

syp2ysy commented 1 month ago

self.layer1, self.layer2, self.layer3, self.layer4 = nn.Sequential(self.dinov2.blocks[:8]), nn.Sequential(self.dinov2.blocks[8:12]), nn.Sequential(self.dinov2.blocks[12:21]), nn.Sequential(self.dinov2.blocks[21:24])