jschwab21 commited on
Commit
40ddb04
·
verified ·
1 Parent(s): a8fc1f2

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +1 -0
video_processing.py CHANGED
@@ -119,6 +119,7 @@ def analyze_scenes(video_path, scenes, description):
119
  positive_similarity = torch.cosine_similarity(image_features, positive_feature.unsqueeze(0)).squeeze().item()
120
  negative_similarities = torch.cosine_similarity(image_features, negative_features).squeeze().mean().item()
121
  scene_prob += positive_similarity - negative_similarities
 
122
 
123
  scene_prob /= len(frames)
124
  scene_duration = convert_timestamp_to_seconds(end_time) - convert_timestamp_to_seconds(start_time)
 
119
  positive_similarity = torch.cosine_similarity(image_features, positive_feature.unsqueeze(0)).squeeze().item()
120
  negative_similarities = torch.cosine_similarity(image_features, negative_features).squeeze().mean().item()
121
  scene_prob += positive_similarity - negative_similarities
122
+ print(classify_frame(frame))
123
 
124
  scene_prob /= len(frames)
125
  scene_duration = convert_timestamp_to_seconds(end_time) - convert_timestamp_to_seconds(start_time)