nam_nguyenhoai_AI
commited on
Commit
·
3d8c5e9
1
Parent(s):
7cb8f79
fix bug
Browse files
app.py
CHANGED
@@ -96,14 +96,14 @@ def make_video(video_path, outdir='./summarized_video',encoder='Kmeans'):
|
|
96 |
|
97 |
number_of_clusters = round(len(features)*0.15)
|
98 |
|
99 |
-
print("
|
100 |
print("Shape of each frame: ", frames[0].shape)
|
101 |
-
print("
|
102 |
-
print("Shape of each
|
103 |
|
104 |
selected_frames = []
|
105 |
if encoder == "Kmeans":
|
106 |
-
selected_frames = kmeans(
|
107 |
elif encoder == "Sum of Squared Difference 01":
|
108 |
selected_frames = tt01(features, 400)
|
109 |
else:
|
|
|
96 |
|
97 |
number_of_clusters = round(len(features)*0.15)
|
98 |
|
99 |
+
print("Total of frames: ", len(final_key_frames))
|
100 |
print("Shape of each frame: ", frames[0].shape)
|
101 |
+
print("Total of clips: ", len(features))
|
102 |
+
print("Shape of each clip: ", features[0].shape)
|
103 |
|
104 |
selected_frames = []
|
105 |
if encoder == "Kmeans":
|
106 |
+
selected_frames = kmeans(number_of_clusters, features)
|
107 |
elif encoder == "Sum of Squared Difference 01":
|
108 |
selected_frames = tt01(features, 400)
|
109 |
else:
|