search

facebookresearch / detr

End-to-End Object Detection with Transformers
Apache License 2.0
13.08k stars 2.37k forks source link

if object query is random, as shown in code, how to evaluate to get a steady result? #597

Open RedBlack888 opened 11 months ago

RedBlack888 commented 11 months ago

if object query is random, as shown in code, how to evaluate to get a steady result when inference? self.query_embed = nn.Embedding(num_queries, hidden_dim)

hs = self.transformer(self.input_proj(src), mask, self.query_embed.weight, pos[-1])[0]

Best.

AjibolaPy commented 9 months ago

It's learnable. I think you'll use the learnt weight on inference