Spaces:
Runtime error
Runtime error
wjf5203
commited on
Commit
•
d565de0
1
Parent(s):
8468984
add video func support
Browse files
app.py
CHANGED
@@ -611,7 +611,7 @@ def segment_video(video, prompt_mode, categoryname, custom_category, expressiong
|
|
611 |
color=(255,255,255),
|
612 |
)
|
613 |
|
614 |
-
ourput_frames.append(ret)
|
615 |
# ret = ret.astype('uint8')
|
616 |
size = (ori_width,ori_height)
|
617 |
print('writing video...')
|
@@ -733,7 +733,7 @@ def segment_video(video, prompt_mode, categoryname, custom_category, expressiong
|
|
733 |
# cv2.imwrite('00020_inst.jpg', cv2.cvtColor(ret, cv2.COLOR_BGR2RGB))
|
734 |
|
735 |
output_vos_results = []
|
736 |
-
output_vos_results.append(ret
|
737 |
|
738 |
|
739 |
|
@@ -839,7 +839,7 @@ def segment_video(video, prompt_mode, categoryname, custom_category, expressiong
|
|
839 |
copyed_img = frame_list[frame_idx]
|
840 |
ret = copyed_img*mask_image_mix_ration
|
841 |
ret = ret.astype('uint8')
|
842 |
-
output_vos_results.append(ret)
|
843 |
|
844 |
size = (ori_width,ori_height)
|
845 |
output_file = "test.mp4"
|
|
|
611 |
color=(255,255,255),
|
612 |
)
|
613 |
|
614 |
+
ourput_frames.append(ret[:,:,::-1])
|
615 |
# ret = ret.astype('uint8')
|
616 |
size = (ori_width,ori_height)
|
617 |
print('writing video...')
|
|
|
733 |
# cv2.imwrite('00020_inst.jpg', cv2.cvtColor(ret, cv2.COLOR_BGR2RGB))
|
734 |
|
735 |
output_vos_results = []
|
736 |
+
output_vos_results.append(ret)
|
737 |
|
738 |
|
739 |
|
|
|
839 |
copyed_img = frame_list[frame_idx]
|
840 |
ret = copyed_img*mask_image_mix_ration
|
841 |
ret = ret.astype('uint8')
|
842 |
+
output_vos_results.append(ret[:,:,::-1])
|
843 |
|
844 |
size = (ori_width,ori_height)
|
845 |
output_file = "test.mp4"
|