feat

utils.py

@@ -68,9 +68,8 @@ def post_process_grounded_object_detection_with_queries(
         prob = p[mask]
         query = q[mask]
         result = {"scores": score, "boxes": box, "queries": query}
-        print('scores: ', score.shape, 'boxes: ', box.shape, 'queries: ', query.shape)
+        assert score.shape[0] == box.shape[0] == query.shape[0], "scores, boxes and queries must have the same length, but got {} and {} and {}".format(score.shape[0], box.shape[0], query.shape[0])
         results.append(result)
-    assert results['scores'].shape == results['boxes'].shape == results['queries'].shape, "scores, boxes and queries must have the same shape, but got {} and {}".format(results['scores'].shape, results['boxes'].shape, results['queries'].shape)
     return results