OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2026-01-15 12:15:15 +00:00
Code Issues Actions Packages Projects 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
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)