simcop2387/faceswap
1
0
Fork 0
mirror of https://github.com/deepfakes/faceswap synced 2025-06-07 10:43:27 -04:00
Code Issues Releases Wiki Activity

manual tool: bugfx: don't error when getting mesh for non-existant face

Browse source
This commit is contained in:
torzdf 2024-04-17 14:16:33 +01:00
parent ec2a95adf6
commit 3f69d9feab
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tools/manual/faceviewer/viewport.py
View file

@ -481,7 +481,7 @@ class Recycler:
asset_type, asset_id)
retval.setdefault(asset_type, []).append(asset_id)
logger.trace("Got mesh: %s", retval) # type:ignore[attr-defined]
logger.info("Got mesh: %s", retval) # type:ignore[attr-defined]
return retval
@ -647,7 +647,8 @@ class VisibleObjects():
y_coord = base_coords[0][1] + (row * self._size)
images.append([self._recycler.get_image((coords[0], y_coord))
for coords in base_coords])
meshes.append([self._recycler.get_mesh(face) for face in self._visible_faces[row]])
meshes.append([{} if face is None else self._recycler.get_mesh(face)
for face in self._visible_faces[row]])
a_images = np.array(images)
a_meshes = np.array(meshes)