feat(example): extract example as png files

Author: AmitMY

README.md

@@ -84,4 +84,12 @@ preserved is the choice of signs.
     howpublished={\url{https://github.com/sign-language-processing/pose-anonymization}},
     year={2024}
 }
-```
+```
+
+
+| Sign   | Original                                   | Anonymized                                     | Transferred                                      |
+|--------|--------------------------------------------|------------------------------------------------|--------------------------------------------------| 
+| Kleine | ![kleine](figures/example/original/kleine.png) | ![kleine](figures/example/anonymized/kleine.png) | ![kleine](figures/example/interpreter/kleine.png) |
+| Kinder | ![kinder](figures/example/original/kinder.png) | ![kinder](figures/example/anonymized/kinder.png) | ![kinder](figures/example/interpreter/kinder.png) |
+| Essen  | ![essen](figures/example/original/essen.png) | ![essen](figures/example/anonymized/essen.png)  | ![essen](figures/example/interpreter/essen.png)   |
+| Pizza  | ![pizza](figures/example/original/pizza.png) | ![pizza](figures/example/anonymized/pizza.png)  | ![pizza](figures/example/interpreter/pizza.png)   |

assets/example/anonymized/essen.png

@@ -14,6 +14,11 @@ ffmpeg -i assets/example/original/kinder.mp4 assets/example/original/kinder.gif
 ffmpeg -i assets/example/original/essen.mp4 assets/example/original/essen.gif -y
 ffmpeg -i assets/example/original/pizza.mp4 assets/example/original/pizza.gif -y
 
+python ./scripts/extract_middle_frame.py --pose=assets/example/original/kleine.pose --output=assets/example/original/kleine.png
+python ./scripts/extract_middle_frame.py --pose=assets/example/original/kinder.pose --output=assets/example/original/kinder.png
+python ./scripts/extract_middle_frame.py --pose=assets/example/original/essen.pose --output=assets/example/original/essen.png
+python ./scripts/extract_middle_frame.py --pose=assets/example/original/pizza.pose --output=assets/example/original/pizza.png
+
 rm assets/example/original/kleine.mp4
 rm assets/example/original/kinder.mp4
 rm assets/example/original/essen.mp4
@@ -36,6 +41,11 @@ ffmpeg -i assets/example/anonymized/kinder.mp4 assets/example/anonymized/kinder.
 ffmpeg -i assets/example/anonymized/essen.mp4 assets/example/anonymized/essen.gif -y
 ffmpeg -i assets/example/anonymized/pizza.mp4 assets/example/anonymized/pizza.gif -y
 
+python ./scripts/extract_middle_frame.py --pose=assets/example/anonymized/kleine.pose --output=assets/example/anonymized/kleine.png
+python ./scripts/extract_middle_frame.py --pose=assets/example/anonymized/kinder.pose --output=assets/example/anonymized/kinder.png
+python ./scripts/extract_middle_frame.py --pose=assets/example/anonymized/essen.pose --output=assets/example/anonymized/essen.png
+python ./scripts/extract_middle_frame.py --pose=assets/example/anonymized/pizza.pose --output=assets/example/anonymized/pizza.png
+
 rm assets/example/anonymized/kleine.mp4
 rm assets/example/anonymized/kinder.mp4
 rm assets/example/anonymized/essen.mp4
@@ -57,6 +67,11 @@ ffmpeg -i assets/example/interpreter/kinder.mp4 assets/example/interpreter/kinde
 ffmpeg -i assets/example/interpreter/essen.mp4 assets/example/interpreter/essen.gif -y
 ffmpeg -i assets/example/interpreter/pizza.mp4 assets/example/interpreter/pizza.gif -y
 
+python ./scripts/extract_middle_frame.py --pose=assets/example/interpreter/kleine.pose --output=assets/example/interpreter/kleine.png
+python ./scripts/extract_middle_frame.py --pose=assets/example/interpreter/kinder.pose --output=assets/example/interpreter/kinder.png
+python ./scripts/extract_middle_frame.py --pose=assets/example/interpreter/essen.pose --output=assets/example/interpreter/essen.png
+python ./scripts/extract_middle_frame.py --pose=assets/example/interpreter/pizza.pose --output=assets/example/interpreter/pizza.png
+
 rm assets/example/interpreter/kleine.mp4
 rm assets/example/interpreter/kinder.mp4
 rm assets/example/interpreter/essen.mp4

assets/example/anonymized/kinder.png

@@ -0,0 +1,24 @@
+import argparse
+
+from pose_format import Pose
+from pose_format.pose_visualizer import PoseVisualizer
+
+
+def main(pose_path: str, output_path: str):
+    with open(pose_path, 'rb') as pose_file:
+        pose = Pose.read(pose_file.read())
+        middle_frame = pose.body.data.shape[0] // 2
+        pose.body = pose.body[middle_frame:middle_frame + 1]
+
+    vis = PoseVisualizer(pose)
+    frame = next(vis.draw(background_color=(255, 255, 255)))
+    vis.save_frame(output_path, frame)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='Extract middle frame from a pose file')
+    parser.add_argument('--pose', type=str, help='Path to the pose file')
+    parser.add_argument('--output', type=str, help='Path to the output file')
+    args = parser.parse_args()
+
+    main(args.pose, args.output)