xmrt commited on
Commit
fdb7efd
1 Parent(s): 6055874

visualization

Browse files
Files changed (1) hide show
  1. main.py +2 -2
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(output_file, cv2.cvtColor(frame[0], cv2.COLOR_BGR2RGB))
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!")