Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-07-11 21:09:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933]add timestamp for movie scene segmentation output

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10269467

    * add timestamp for movie scene segmentation output
This commit is contained in:
shuying.shu
2022-09-27 22:15:24 +08:00
committed by yingda.chen
parent e90ff9e479
commit b98114367b
3 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
modelscope/models/audio/tts/models/datasets/__init__.py Normal file → Executable file
View File

12
modelscope/models/cv/movie_scene_segmentation/utils/save_op.py
View File

@@ -26,7 +26,8 @@ def pred2scene(shot2keyf, anno_dict):
for scene_ind, scene_item in enumerate(scene_list):
scene_dict_lst.append({
'shot': pair_list[scene_ind],
'frame': scene_item
'frame': scene_item[0],
'timestamp': scene_item[1]
})
return scene_dict_lst, scene_list
@@ -42,8 +43,8 @@ def scene2video(source_movie_fn, scene_list, thres):
for scene_ind, scene_item in tqdm(enumerate(scene_list)):
scene = str(scene_ind).zfill(4)
start_frame = int(scene_item[0])
end_frame = int(scene_item[1])
start_frame = int(scene_item[0][0])
end_frame = int(scene_item[0][1])
start_time, end_time = start_frame / fps, end_frame / fps
duration_time = end_time - start_time
out_video_fn = os.path.join(out_video_dir_fn,
@@ -71,7 +72,10 @@ def get_demo_scene_list(shot2keyf, anno_dict):
start_shot, end_shot = int(pair[0]), int(pair[-1])
start_frame = shot2keyf[start_shot].split(' ')[0]
end_frame = shot2keyf[end_shot].split(' ')[1]
scene_list.append((start_frame, end_frame))
start_timestamp = shot2keyf[start_shot].split(' ')[-2]
end_timestamp = shot2keyf[end_shot].split(' ')[-1]
scene_list.append([[start_frame, end_frame],
[start_timestamp, end_timestamp]])
return scene_list, pair_list

3
modelscope/outputs.py
View File

@@ -303,7 +303,8 @@ TASK_OUTPUTS = {
# [
# {
# "shot": [0,1,2],
# "frame": [start_frame, end_frame]
# "frame": [start_frame, end_frame],
# "timestamp": [start_timestamp, end_timestamp] # ['00:00:01.133', '00:00:02.245']
# }
# ]
#