Files
object-detection-flask-opencv/camera_settings.py
Diego Inácio 2840ecdcb8 Upload modules
Camera settings and Object detection
2020-02-02 16:55:32 -03:00

34 lines
917 B
Python

import os
import cv2
attrib_list = {
'exposure': cv2.CAP_PROP_EXPOSURE,
'contrast': cv2.CAP_PROP_CONTRAST
}
def check_settings():
VIDEO_CHECK = cv2.VideoCapture(0)
if not os.path.exists('camera_settings.log'):
f = open('camera_settings.log', 'w')
for attrib, index in attrib_list.items():
f.writelines(f'{attrib} = {VIDEO_CHECK.get(index)}\n')
f.close()
else:
f = open('camera_settings.log', 'r')
lines = f.read().split('\n')
for line in lines:
attrib = line.split(' = ')
if attrib[0] in attrib_list.keys():
VIDEO_CHECK.set(attrib_list[attrib[0]], eval(attrib[1]))
f.close()
print('*'*28)
print('* Checking camera settings *')
print('*'*28)
for attrib, index in attrib_list.items():
print(f'{attrib} = {VIDEO_CHECK.get(index)}')
VIDEO_CHECK.release()