diff --git a/lib/convert.py b/lib/convert.py index c09703e5..3f782f2e 100644 --- a/lib/convert.py +++ b/lib/convert.py @@ -86,6 +86,8 @@ class Converter(): in_queue.put(items) break + if isinstance(items, dict): + items = [items] for item in items: logger.trace("Patch queue got: '%s'", item["filename"]) try: diff --git a/tools/preview.py b/tools/preview.py index 78a925df..570ed26b 100644 --- a/tools/preview.py +++ b/tools/preview.py @@ -245,13 +245,14 @@ class Samples(): idx = 0 while idx < self.sample_size: logger.debug("Predicting face %s of %s", idx + 1, self.sample_size) - item = self.predictor.out_queue.get() - if item == "EOF": + items = self.predictor.out_queue.get() + if items == "EOF": logger.debug("Received EOF") break - self.predicted_images.append(item) - logger.debug("Predicted face %s of %s", idx + 1, self.sample_size) - idx += 1 + for item in items: + self.predicted_images.append(item) + logger.debug("Predicted face %s of %s", idx + 1, self.sample_size) + idx += 1 logger.debug("Predicted faces")