search

WangYueFt / detr3d

MIT License
792 stars 155 forks source link

what is meaning of self.query_embedding = nn.Embedding(self.num_query, self.embed_dims * 2) #72

Open hitbuyi opened 7 months ago

hitbuyi commented 7 months ago

in detr3d_head.py, 

       if not self.as_two_stage:
            self.query_embedding = nn.Embedding(self.num_query,
                                                self.embed_dims * 2)

why self.embed_dims multiplfies *2? what's the meaning here?

Raiden-cn commented 2 months ago

I guess it initializes object query like DETR.