fight-object_detection / full_project.py
KillD00zer's picture
Upload 9 files
da07a7d verified
raw
history blame contribute delete
490 Bytes
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)