diff --git a/feat/facepose_detectors/img2pose/img2pose_test.py b/feat/facepose_detectors/img2pose/img2pose_test.py index 3671b117..e7e81801 100644 --- a/feat/facepose_detectors/img2pose/img2pose_test.py +++ b/feat/facepose_detectors/img2pose/img2pose_test.py @@ -183,6 +183,12 @@ def predict(self, img, border_size=0, scale=1.0, euler=True): dict: A dictionary of bboxes and poses """ + # For device='mps' + # Uncommenting this line at least gets img2pose running but errors with + # Error: command buffer exited with error status. + # The Metal Performance Shaders operations encoded on it may not have completed. + + # img = img.to(self.device) # Obtain prediction pred = self.model.predict([img])[0]