Violence_Detector / src /preprocess.py
muskan19's picture
Upload preprocess.py
9f9536d verified
raw
history blame contribute delete
288 Bytes
# preprocess.py placeholder
import cv2
import numpy as np
IMG_SIZE = (224, 224)
def preprocess_frame(frame):
frame = cv2.resize(frame, IMG_SIZE)
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
frame = np.expand_dims(frame, axis=0)
frame = frame / 255.0
return frame