Skip to content

Commit

Permalink
Update config.py
Browse files Browse the repository at this point in the history
Signed-off-by: Bubbles The Dev <152947339+KernFerm@users.noreply.github.com>
  • Loading branch information
KernFerm authored Sep 6, 2024
1 parent 9937842 commit 37aaddb
Showing 1 changed file with 22 additions and 2 deletions.
24 changes: 22 additions & 2 deletions config.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,24 @@
# Configuration for BetterCam Screen Capture and YOLO model
screenWidth = 320 # Recommended screen width
screenHeight = 320 # Recommended screen height

# Screen Capture Settings
screenWidth = 480 # Updated screen width for better resolution
screenHeight = 480 # Updated screen height for better resolution

# Object Detection Settings
confidenceThreshold = 0.5 # Confidence threshold for object detection
nmsThreshold = 0.4 # Non-max suppression threshold to filter overlapping boxes

# YOLO Model Settings
modelType = 'onnx' # Choose 'torch' or 'onnx' based on the model you want to load
torchModelPath = 'models/fn_v5.pt' # Path to YOLOv5 PyTorch model
onnxModelPath = 'models/fn_v5.onnx' # Path to YOLOv5 ONNX model

# 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

# Colors for Bounding Boxes
boundingBoxColor = (0, 255, 0) # Default bounding box color in BGR format
highlightColor = (0, 0, 255) # Color for highlighted objects

0 comments on commit 37aaddb

Please sign in to comment.