jschwab21 commited on
Commit
91fc83c
·
verified ·
1 Parent(s): e55284f

Update video_processing.py

Browse files
Files changed (1) hide show
  1. 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=60):
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(),