default_config.py 847 B

12345678910111213141516171819202122232425
  1. import cv2
  2. import os
  3. os.environ["QT_QPA_PLATFORM"] = "xcb"
  4. MARKER_SIZE = 5 # centimeters
  5. # LỰa chọn độ phân giải bằng cách thay đổi set_resolution
  6. set_resolution = 'original'
  7. RESOLUTION_STANDARDS = {'540p': 540, '720p': 720, '1080p': 1080, 'original': None}
  8. # Thay đổi độ phân giải phù hợp với yêu cầu ban đầu
  9. def change_resolution(capture: cv2.VideoCapture, new_height):
  10. old_height = int(capture.get(cv2.CAP_PROP_FRAME_HEIGHT))
  11. old_width = int(capture.get(cv2.CAP_PROP_FRAME_WIDTH))
  12. if new_height is not None:
  13. new_width = int(old_width* new_height/old_height)
  14. capture.set(cv2.CAP_PROP_FRAME_WIDTH, new_width)
  15. capture.set(cv2.CAP_PROP_FRAME_HEIGHT, new_height)
  16. else:
  17. new_width = old_width
  18. new_height = old_height
  19. return (new_width, new_height)