support coco evaluation (#149)

2023-06-17 02:31:07 -07:00
parent 4e6f23d35c
commit 16292e162d
6 changed files with 521 additions and 15 deletions
--- a/groundingdino/models/GroundingDINO/groundingdino.py
+++ b/groundingdino/models/GroundingDINO/groundingdino.py
@@ -228,7 +228,6 @@ class GroundingDINO(nn.Module):
            captions = kw["captions"]
        else:
            captions = [t["caption"] for t in targets]
-        len(captions)

        # encoder texts
        tokenized = self.tokenizer(captions, padding="longest", return_tensors="pt").to(