Spaces:
Runtime error
Runtime error
desco
Browse files
maskrcnn_benchmark/engine/predictor_glip.py
CHANGED
@@ -139,7 +139,8 @@ class GLIPDemo(object):
|
|
139 |
specified_tokens = None,
|
140 |
**kwargs):
|
141 |
original_caption = original_caption.lower()
|
142 |
-
specified_tokens
|
|
|
143 |
|
144 |
predictions = self.compute_prediction(original_image, original_caption, specified_tokens=specified_tokens)
|
145 |
top_predictions = self._post_process(predictions, thresh)
|
|
|
139 |
specified_tokens = None,
|
140 |
**kwargs):
|
141 |
original_caption = original_caption.lower()
|
142 |
+
if specified_tokens is not None:
|
143 |
+
specified_tokens = [token.lower() for token in specified_tokens]
|
144 |
|
145 |
predictions = self.compute_prediction(original_image, original_caption, specified_tokens=specified_tokens)
|
146 |
top_predictions = self._post_process(predictions, thresh)
|