-
-
Notifications
You must be signed in to change notification settings - Fork 3
/
config.py
47 lines (37 loc) · 2.14 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# Configuration for BetterCam Screen Capture and YOLO model
# Screen Capture Settings
screenWidth = 480 # Updated screen width for better resolution - (recommended 480)
screenHeight = 480 # Updated screen height for better resolution - (recommended 480)
# Object Detection Settings
confidenceThreshold = 0.5 # Confidence threshold for object detection
nmsThreshold = 0.4 # Non-max suppression threshold to filter overlapping boxes
# YOLO Model Selection
# Choose the type of model you want to use: 'torch', 'onnx', or 'engine'
# 'torch' is for PyTorch models (.pt), 'onnx' is for ONNX models (.onnx), and 'engine' is for TensorRT models (.engine)
modelType = 'onnx' # Example: Set to 'torch' for PyTorch models, 'onnx' for ONNX models, 'engine' for TensorRT models
# Model Paths (YOLOv5 and YOLOv8)
# Uncomment the model path corresponding to the version of YOLO you want to use
# YOLOv5 PyTorch and ONNX models
torchModelPath = 'models/fn_v5.pt' # YOLOv5 PyTorch model path (.pt)
onnxModelPath = 'models/fn_v5.onnx' # YOLOv5 ONNX model path (.onnx)
# YOLOv8 PyTorch and ONNX models
# torchModelPath = 'models/fn_v8.pt' # YOLOv8 PyTorch model path (.pt)
# onnxModelPath = 'models/fn_v8.onnx' # YOLOv8 ONNX model path (.onnx)
# TensorRT Model
# tensorrtModelPath = 'models/fn_model.engine' # Path to TensorRT model (.engine)
# BetterCam Settings
targetFPS = 60 # Frames per second for capturing
maxBufferLen = 512 # Max buffer length for storing frames
region = None # Region for capture (set to None for full screen)
useNvidiaGPU = True # Set to True to enable GPU acceleration if available
monitorIdx = 0 # Index for multi-monitor support, set 0 for primary monitor
# Colors for Bounding Boxes
boundingBoxColor = (0, 255, 0) # Default bounding box color in BGR format
highlightColor = (0, 0, 255) # Color for highlighted objects
# Overlay Settings
overlayWidth = 1920 # Overlay window width
overlayHeight = 1080 # Overlay window height
overlayAlpha = 0.6 # Transparency of overlay
# GPU Support
useCuda = True # Enable CUDA support if available
useDirectML = False # Set to True to enable DirectML for AMD GPUs