desco

maskrcnn_benchmark/engine/predictor_glip.py

@@ -139,7 +139,8 @@ class GLIPDemo(object):
             specified_tokens = None,
             **kwargs):
         original_caption = original_caption.lower()
-        specified_tokens = [token.lower() for token in specified_tokens]
+        if specified_tokens is not None:
+            specified_tokens = [token.lower() for token in specified_tokens]
 
         predictions = self.compute_prediction(original_image, original_caption, specified_tokens=specified_tokens)
         top_predictions = self._post_process(predictions, thresh)