Cookies_Github_mirror/Nugget
1
0
Fork 0
mirror of https://github.com/leminlimez/Nugget.git synced 2025-04-08 04:23:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add 300 fps limit

Browse Source
This commit is contained in:
leminlimez
2025-03-27 12:54:09 -04:00
parent cc79e18df8
commit 2982e1d186
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
controllers/video_handler.py
View File

@@ -43,6 +43,10 @@ def get_thumbnail_from_contents(contents: bytes, output_file: str = None):
def create_caml(video_path: str, output_file: str, update_label=lambda x: None):
cam = cv2.VideoCapture(video_path)
assets_path = os.path.join(output_file, "assets")
frame_count = int(cam.get(cv2.CAP_PROP_FRAME_COUNT))
FRAME_LIMIT = 300
if frame_count > FRAME_LIMIT:
raise Exception(f"Videos must be under {FRAME_LIMIT} fps to loop. Either reduce the frame rate or make it shorter.")
try:
# creating a folder named data
if not os.path.exists(assets_path):
@@ -59,7 +63,6 @@ def create_caml(video_path: str, output_file: str, update_label=lambda x: None):
with open(os.path.join(output_file, "main.caml"), "w") as caml:
# write caml header
fps = cam.get(cv2.CAP_PROP_FPS)
frame_count = int(cam.get(cv2.CAP_PROP_FRAME_COUNT))
duration = frame_count / fps
caml.write(f"""<?xml version="1.0" encoding="UTF-8"?>