xmrt commited on
Commit
fd8361a
·
1 Parent(s): d7ba715

height in playable and kptthresh

Browse files
Files changed (1) hide show
  1. main.py +5 -2
main.py CHANGED
@@ -126,7 +126,7 @@ def run():
126
  webcam = gr.Interface(
127
  fn=infer,
128
  inputs= [gr.Video(source="webcam", height=512), check_web],
129
- outputs = [gr.Video(format='mp4'), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512)],
130
  title = 'Pose estimation',
131
  description = 'Pose estimation on video',
132
  allow_flagging=False
@@ -135,7 +135,7 @@ def run():
135
  file = gr.Interface(
136
  infer,
137
  inputs = [gr.Video(source="upload", height=512), check_file],
138
- outputs = [gr.Video(format='mp4'), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512)],
139
  allow_flagging=False
140
  )
141
 
@@ -148,6 +148,9 @@ def run():
148
 
149
 
150
  if __name__ == "__main__":
 
 
 
151
  run()
152
 
153
  # https://github.com/open-mmlab/mmpose/tree/dev-1.x/configs/body_3d_keypoint/pose_lift
 
126
  webcam = gr.Interface(
127
  fn=infer,
128
  inputs= [gr.Video(source="webcam", height=512), check_web],
129
+ outputs = [gr.Video(format='mp4', height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512)],
130
  title = 'Pose estimation',
131
  description = 'Pose estimation on video',
132
  allow_flagging=False
 
135
  file = gr.Interface(
136
  infer,
137
  inputs = [gr.Video(source="upload", height=512), check_file],
138
+ outputs = [gr.Video(format='mp4', height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512), gr.PlayableVideo(height=512)],
139
  allow_flagging=False
140
  )
141
 
 
148
 
149
 
150
  if __name__ == "__main__":
151
+ description = """
152
+ \n\nHere you can upload videos or record one with your webcam and track objects or detect bodyposes in 2d and 3d.
153
+ """
154
  run()
155
 
156
  # https://github.com/open-mmlab/mmpose/tree/dev-1.x/configs/body_3d_keypoint/pose_lift