Update box_utils.py

box_utils.py

@@ -106,7 +106,7 @@ def postprocess(matches, prompt_labels, img_matches=None):
                                     m['logit'], [1] *
                                     len(m['box_id']),
                                     list(np.array(m['cls_emb'])))))
-        if img_matches is not None k in img_matches:
+        if img_matches is not None and k in img_matches:
             img_m = img_matches_[k]
             # and also those non-TopK hits and those non-topk are not anticipating training
             boxes += [i for i in map(list, zip(img_m['box_id'], img_m['cx'], img_m['cy'], img_m['w'], img_m['h'],