Spaces:
Sleeping
Sleeping
Update video_processing.py
Browse files- 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)
|