GithubStar/faceswap
1
0
Fork 0
mirror of https://github.com/deepfakes/faceswap.git synced 2026-10-27 04:19:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed convert bug in preview (#846)

Browse Source
This commit is contained in:
kilroythethird
2019-08-27 14:58:24 +01:00
committed by torzdf
parent db60f6037d
commit e666384792
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/convert.py
+2
View File
@@ -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:
tools/preview.py
+6 -5
View File
@@ -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")