shengShenLan commited on
Commit
101dba8
1 Parent(s): 47c976d
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -32,10 +32,10 @@ def video_processor(video):
32
  height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
33
  fps = cap.get(cv2.CAP_PROP_FPS)
34
 
35
-
36
  # 创建一个VideoWriter对象,输出为mp4格式
37
  fourcc = cv2.VideoWriter_fourcc(*'avc1')
38
- out = cv2.VideoWriter('output.mp4', fourcc, fps, (width, height))
39
 
40
  # 创建一个图形对象
41
  fig, ax = plt.subplots(figsize=(3, 3))
@@ -117,7 +117,9 @@ def video_processor(video):
117
  cap.release()
118
  # 返回输出视频的路径
119
 
120
- return 'output.mp4'
 
 
121
 
122
  import gradio as gr
123
 
 
32
  height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
33
  fps = cap.get(cv2.CAP_PROP_FPS)
34
 
35
+ d = os.path.dirname(video)
36
  # 创建一个VideoWriter对象,输出为mp4格式
37
  fourcc = cv2.VideoWriter_fourcc(*'avc1')
38
+ out = cv2.VideoWriter(os.path.join(d,'output.mp4'), fourcc, fps, (width, height))
39
 
40
  # 创建一个图形对象
41
  fig, ax = plt.subplots(figsize=(3, 3))
 
117
  cap.release()
118
  # 返回输出视频的路径
119
 
120
+ print(video)
121
+ print(os.path.join(d,'output.mp4'))
122
+ return os.path.join(d,'output.mp4')
123
 
124
  import gradio as gr
125