visualization
Browse files
main.py
CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
|
|
7 |
import numpy as np
|
8 |
import cv2
|
9 |
|
10 |
-
inferencer = MMPoseInferencer('human')
|
11 |
print("[INFO]: Downloaded models!")
|
12 |
|
13 |
|
@@ -33,7 +33,7 @@ def poses(photo):
|
|
33 |
frame = result["visualization"]
|
34 |
print(frame)
|
35 |
print(type(frame))
|
36 |
-
out_writer.write(
|
37 |
|
38 |
print(os.listdir())
|
39 |
print("[INFO]: Visualizing results!")
|
|
|
7 |
import numpy as np
|
8 |
import cv2
|
9 |
|
10 |
+
inferencer = MMPoseInferencer('human') # 'hand'
|
11 |
print("[INFO]: Downloaded models!")
|
12 |
|
13 |
|
|
|
33 |
frame = result["visualization"]
|
34 |
print(frame)
|
35 |
print(type(frame))
|
36 |
+
out_writer.write(cv2.cvtColor(frame[0], cv2.COLOR_BGR2RGB))
|
37 |
|
38 |
print(os.listdir())
|
39 |
print("[INFO]: Visualizing results!")
|