fffiloni commited on
Commit
3e4f568
1 Parent(s): 6606e10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -45,15 +45,15 @@ def get_frames(video_in):
45
 
46
  return frames, fps
47
 
48
- def get_canny_filter(i):
49
  image = Image.open(i)
50
 
51
  #image = np.array(image)
52
 
53
  image = openpose(image)
54
  #image = Image.fromarray(image)
55
- image.save("canny_frame_" + str(i) + ".jpeg")
56
- return "canny_frame_" + str(i) + ".jpeg"
57
 
58
  def create_video(frames, fps, type):
59
  print("building video result")
@@ -86,7 +86,7 @@ def infer(video_in):
86
  print("set stop frames to: " + str(n_frame))
87
 
88
  for i in frames_list[0:int(n_frame)]:
89
- canny_frame = get_canny_filter(i)
90
  result_frames.append(canny_frame)
91
  print("frame " + i + "/" + str(n_frame) + ": done;")
92
 
@@ -109,7 +109,7 @@ title="""
109
  "
110
  >
111
  <h1 style="font-weight: 600; margin-bottom: 7px;">
112
- Video to Canny Edge
113
  </h1>
114
  </div>
115
 
 
45
 
46
  return frames, fps
47
 
48
+ def get_openpose_filter(i):
49
  image = Image.open(i)
50
 
51
  #image = np.array(image)
52
 
53
  image = openpose(image)
54
  #image = Image.fromarray(image)
55
+ image.save("openpose_frame_" + str(i) + ".jpeg")
56
+ return "openpose_frame_" + str(i) + ".jpeg"
57
 
58
  def create_video(frames, fps, type):
59
  print("building video result")
 
86
  print("set stop frames to: " + str(n_frame))
87
 
88
  for i in frames_list[0:int(n_frame)]:
89
+ openpose_frame = get_openpose_filter(i)
90
  result_frames.append(canny_frame)
91
  print("frame " + i + "/" + str(n_frame) + ": done;")
92
 
 
109
  "
110
  >
111
  <h1 style="font-weight: 600; margin-bottom: 7px;">
112
+ Video to OpenPose
113
  </h1>
114
  </div>
115