xmrt commited on
Commit
abeb59d
1 Parent(s): a815c05

visualization

Browse files
Files changed (1) hide show
  1. main.py +8 -2
main.py CHANGED
@@ -13,13 +13,19 @@ print("[INFO]: Downloaded models!")
13
  def poses(photo):
14
  result_generator = inferencer(photo,
15
  vis_out_dir =".",
16
- return_vis=True)
 
 
17
  print("[INFO]: Visualizing results!")
18
 
19
  # The MMPoseInferencer API employs a lazy inference approach,
20
  # creating a prediction generator when given input
21
  result = next(result_generator)
22
- print(type(result['visualization']))
 
 
 
 
23
  return "000000.jpg"
24
 
25
  # # specify detection model by alias
 
13
  def poses(photo):
14
  result_generator = inferencer(photo,
15
  vis_out_dir =".",
16
+ return_vis=True,
17
+ thickness=2)
18
+
19
  print("[INFO]: Visualizing results!")
20
 
21
  # The MMPoseInferencer API employs a lazy inference approach,
22
  # creating a prediction generator when given input
23
  result = next(result_generator)
24
+
25
+ print("[INFO]: Type of vis is ", type(result['visualization']))
26
+
27
+ print("[INFO]: Vis is ", type(result['visualization']))
28
+
29
  return "000000.jpg"
30
 
31
  # # specify detection model by alias