diaojiaolou
/
Deep-Live-Cam
mirror of https://github.com/hacksider/Deep-Live-Cam.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(): adding cv2_enumerate_cameras to get camera_names

pull/686/head
GhoulBoii 2024-10-02 23:56:07 +05:30
parent fa904a7bf0
commit 54b52dfbc8
No known key found for this signature in database
GPG Key ID: F443BDA92506073F
2 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
modules/ui.py
View File

@ -3,6 +3,7 @@ import webbrowser
import customtkinter as ctk import customtkinter as ctk
from typing import Callable, Tuple from typing import Callable, Tuple
import cv2 import cv2
from cv2_enumerate_cameras import enumerate_cameras
from PIL import Image, ImageOps from PIL import Image, ImageOps
import tkinterdnd2 as tkdnd import tkinterdnd2 as tkdnd
import time import time
@ -408,7 +409,7 @@ def create_root(
camera_label.place(relx=0.4, rely=0.86, relwidth=0.2, relheight=0.05) camera_label.place(relx=0.4, rely=0.86, relwidth=0.2, relheight=0.05)
available_cameras = get_available_cameras() available_cameras = get_available_cameras()
# Convert camera indices to strings for CTkOptionMenu # Convert camera indices to strings for CTkOptionMenu
available_camera_strings = [str(cam) for cam in available_cameras] available_camera_indices, available_camera_strings = available_cameras
camera_variable = ctk.StringVar( camera_variable = ctk.StringVar(
value=available_camera_strings[0] value=available_camera_strings[0]
if available_camera_strings if available_camera_strings
@ -424,6 +425,12 @@ def create_root(
button_frame, button_frame,
text="Live", text="Live",
cursor="hand2", cursor="hand2",
command=lambda: webcam_preview(
root,
available_camera_indices[
available_camera_strings.index(camera_variable.get())
],
),
) )
live_button.pack(side="left", padx=10, expand=True) live_button.pack(side="left", padx=10, expand=True)
@ -1011,14 +1018,17 @@ def webcam_preview(root: ctk.CTk, camera_index: int):
def get_available_cameras(): def get_available_cameras():
"""Returns a list of available camera indices.""" """Returns a list of available camera names and indices."""
available_cameras = [] camera_indices = []
for index in range(10): # Check for cameras with index 0 to 9 camera_names = []
camera = cv2.VideoCapture(index)
if camera.isOpened(): for camera in enumerate_cameras():
available_cameras.append(index) cap = cv2.VideoCapture(camera.index)
camera.release() if cap.isOpened():
return available_cameras camera_indices.append(camera.index)
camera_names.append(camera.name)
cap.release()
return (camera_indices, camera_names)
# Add this function to update the opacity value # Add this function to update the opacity value

1
requirements.txt
View File

@ -2,6 +2,7 @@
numpy>=1.23.5,<2 numpy>=1.23.5,<2
opencv-python==4.8.1.78 opencv-python==4.8.1.78
cv2_enumerate_cameras==1.1.15
onnx==1.16.0 onnx==1.16.0
insightface==0.7.3 insightface==0.7.3
psutil==5.9.8 psutil==5.9.8