You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
yolov8/MTSP-main/utils/AtestCamera.py

28 lines
682 B

# -*- coding: utf-8 -*-
# @Author : pan
import cv2
class Camera:
def __init__(self, cam_preset_num=5):
self.cam_preset_num = cam_preset_num
def get_cam_num(self):
cnt = 0
devices = []
for device in range(0, self.cam_preset_num):
stream = cv2.VideoCapture(device, cv2.CAP_DSHOW)
grabbed = stream.grab()
stream.release()
if not grabbed:
continue
else:
cnt = cnt + 1
devices.append(device)
return cnt, devices
if __name__ == '__main__':
cam = Camera()
cam_num, devices = cam.get_cam_num()
print(cam_num, devices)