nam_nguyenhoai_AI commited on
Commit
3d8c5e9
·
1 Parent(s): 7cb8f79
Files changed (1) hide show
  1. app.py +4 -4
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("Number of frames: ", len(final_key_frames))
100
  print("Shape of each frame: ", frames[0].shape)
101
- print("Number of features: ", len(features))
102
- print("Shape of each feature: ", features[0].shape)
103
 
104
  selected_frames = []
105
  if encoder == "Kmeans":
106
- selected_frames = kmeans(features, number_of_clusters)
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: