|
|
|
@ -1354,7 +1354,7 @@ class ModelProcessor: |
|
|
|
|
|
|
|
|
|
|
|
# 使用与CPU版本相同的后续处理 |
|
|
|
# 使用与CPU版本相同的后续处理 |
|
|
|
shrunk_visibility = self._shrink_face_visibility(face_visible.cpu().numpy(), 12) |
|
|
|
shrunk_visibility = self._shrink_face_visibility(face_visible.cpu().numpy(), 12) |
|
|
|
expanded_visibility = self._expand_face_visibility(face_visible.cpu().numpy(), 0) |
|
|
|
expanded_visibility = self._expand_face_visibility(face_visible.cpu().numpy(), 30) |
|
|
|
shrunk_visibility2 = self._shrink_face_visibility(face_visible.cpu().numpy(), 50) |
|
|
|
shrunk_visibility2 = self._shrink_face_visibility(face_visible.cpu().numpy(), 50) |
|
|
|
expanded_edge = expanded_visibility & ~shrunk_visibility2 |
|
|
|
expanded_edge = expanded_visibility & ~shrunk_visibility2 |
|
|
|
delete_edge = face_visible.cpu().numpy() & ~shrunk_visibility |
|
|
|
delete_edge = face_visible.cpu().numpy() & ~shrunk_visibility |
|
|
|
|