File size: 490 Bytes
da07a7d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import cv2
import numpy as np
import os
from ultralytics import YOLO
import time
import tensorflow as tf
from frame_slicer import extract_video_frames
import matplotlib.pyplot as plt
from Fight_detec_func import fight_detec
from objec_detect_yolo import detection
# Entry point
path0 = input("Enter the local path : ")
path = path0.strip('"') # Remove extra quotes if copied from Windows
print(f"[INFO] Loading video: {path}")
fight_detec(path)
detection(path)
|