xmrt commited on
Commit
6055874
·
1 Parent(s): f8466d6

visualization

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -33,7 +33,7 @@ def poses(photo):
33
  frame = result["visualization"]
34
  print(frame)
35
  print(type(frame))
36
- out_writer.write(frame[0])
37
 
38
  print(os.listdir())
39
  print("[INFO]: Visualizing results!")
 
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!")