Spaces:
Sleeping
Sleeping
Update video_processing.py
Browse files- video_processing.py +3 -2
video_processing.py
CHANGED
@@ -93,7 +93,7 @@ def extract_frames(video, start_time, end_time):
|
|
93 |
frames.append(frame)
|
94 |
return frames
|
95 |
|
96 |
-
def analyze_scenes(video_path, scenes, description, batch_size=
|
97 |
scene_scores = []
|
98 |
negative_descriptions = [
|
99 |
"black screen",
|
@@ -101,7 +101,8 @@ def analyze_scenes(video_path, scenes, description, batch_size=60):
|
|
101 |
"dark scene without much contrast",
|
102 |
"No people are in this scene",
|
103 |
"A still shot of natural scenery",
|
104 |
-
"Still-camera shot of a person's face"
|
|
|
105 |
]
|
106 |
preprocess = transforms.Compose([
|
107 |
transforms.ToTensor(),
|
|
|
93 |
frames.append(frame)
|
94 |
return frames
|
95 |
|
96 |
+
def analyze_scenes(video_path, scenes, description, batch_size=65):
|
97 |
scene_scores = []
|
98 |
negative_descriptions = [
|
99 |
"black screen",
|
|
|
101 |
"dark scene without much contrast",
|
102 |
"No people are in this scene",
|
103 |
"A still shot of natural scenery",
|
104 |
+
"Still-camera shot of a person's face",
|
105 |
+
"One lazy dog on a log"
|
106 |
]
|
107 |
preprocess = transforms.Compose([
|
108 |
transforms.ToTensor(),
|