jschwab21 commited on
Commit
a82f04a
·
verified ·
1 Parent(s): d4f2cec

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +1 -1
video_processing.py CHANGED
@@ -103,7 +103,7 @@ def analyze_scenes(video_path, scenes, description, batch_size=6):
103
  #"A still shot of natural scenery",
104
  #"Still-camera shot of a person's face"
105
  ]
106
- preprocess = transforms.Compose([
107
  transforms.ToTensor(), # Convert numpy arrays directly to tensors
108
  transforms.Resize((224, 224)), # Resize the tensor to fit model input
109
  transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) # Normalize the tensor
 
103
  #"A still shot of natural scenery",
104
  #"Still-camera shot of a person's face"
105
  ]
106
+ preprocess = transforms.Compose([
107
  transforms.ToTensor(), # Convert numpy arrays directly to tensors
108
  transforms.Resize((224, 224)), # Resize the tensor to fit model input
109
  transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) # Normalize the tensor