Commit
·
228a5fd
1
Parent(s):
6b3879a
debug
Browse files- hugging_face/app.py +3 -4
hugging_face/app.py
CHANGED
|
@@ -116,9 +116,8 @@ def get_frames_from_video(video_input, video_state):
|
|
| 116 |
user_name = time.time()
|
| 117 |
|
| 118 |
# extract Audio
|
| 119 |
-
audio_path = "audio.wav"
|
| 120 |
-
audio_path = video_input.replace(".mp4", "_audio.wav")
|
| 121 |
try:
|
|
|
|
| 122 |
ffmpeg.input(video_path).output(audio_path, format='wav', acodec='pcm_s16le', ac=2, ar='44100').run(overwrite_output=True, quiet=True)
|
| 123 |
except Exception as e:
|
| 124 |
print(f"Audio extraction error: {str(e)}")
|
|
@@ -259,7 +258,7 @@ def show_mask(video_state, interactive_state, mask_dropdown):
|
|
| 259 |
|
| 260 |
# image matting
|
| 261 |
def image_matting(video_state, interactive_state, mask_dropdown, erode_kernel_size, dilate_kernel_size, refine_iter):
|
| 262 |
-
matanyone_processor.
|
| 263 |
if interactive_state["track_end_number"]:
|
| 264 |
following_frames = video_state["origin_images"][video_state["select_frame_number"]:interactive_state["track_end_number"]]
|
| 265 |
else:
|
|
@@ -287,7 +286,7 @@ def image_matting(video_state, interactive_state, mask_dropdown, erode_kernel_si
|
|
| 287 |
|
| 288 |
# video matting
|
| 289 |
def video_matting(video_state, interactive_state, mask_dropdown, erode_kernel_size, dilate_kernel_size):
|
| 290 |
-
matanyone_processor.
|
| 291 |
if interactive_state["track_end_number"]:
|
| 292 |
following_frames = video_state["origin_images"][video_state["select_frame_number"]:interactive_state["track_end_number"]]
|
| 293 |
else:
|
|
|
|
| 116 |
user_name = time.time()
|
| 117 |
|
| 118 |
# extract Audio
|
|
|
|
|
|
|
| 119 |
try:
|
| 120 |
+
audio_path = video_input.replace(".mp4", "_audio.wav")
|
| 121 |
ffmpeg.input(video_path).output(audio_path, format='wav', acodec='pcm_s16le', ac=2, ar='44100').run(overwrite_output=True, quiet=True)
|
| 122 |
except Exception as e:
|
| 123 |
print(f"Audio extraction error: {str(e)}")
|
|
|
|
| 258 |
|
| 259 |
# image matting
|
| 260 |
def image_matting(video_state, interactive_state, mask_dropdown, erode_kernel_size, dilate_kernel_size, refine_iter):
|
| 261 |
+
matanyone_processor = InferenceCore(matanyone_model, cfg=matanyone_model.cfg)
|
| 262 |
if interactive_state["track_end_number"]:
|
| 263 |
following_frames = video_state["origin_images"][video_state["select_frame_number"]:interactive_state["track_end_number"]]
|
| 264 |
else:
|
|
|
|
| 286 |
|
| 287 |
# video matting
|
| 288 |
def video_matting(video_state, interactive_state, mask_dropdown, erode_kernel_size, dilate_kernel_size):
|
| 289 |
+
matanyone_processor = InferenceCore(matanyone_model, cfg=matanyone_model.cfg)
|
| 290 |
if interactive_state["track_end_number"]:
|
| 291 |
following_frames = video_state["origin_images"][video_state["select_frame_number"]:interactive_state["track_end_number"]]
|
| 292 |
else:
|