File size: 839 Bytes
bab4aa8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
63f174e
bab4aa8
 
 
 
 
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
import os

# Model settings
MODEL_ID = "Xenova/clip-vit-base-patch32"
PROCESSOR_ID = "openai/clip-vit-base-patch32"
DEVICE = "cpu"

# ONNX options
INTRA_OP_NUM_THREADS = 4
INTER_OP_NUM_THREADS = 4

# Storage paths
IMAGES_DIR = "captured_images"

# Vector Search
SIMILARITY_THRESHOLD = 0.65

# Object Detection (YOLOv8n ONNX)
ENABLE_OBJECT_DETECTION = os.getenv("ENABLE_OBJECT_DETECTION", "false").lower() == "true"
YOLO_MODEL_ID = os.getenv("YOLO_MODEL_ID", "Kalray/yolov8")
YOLO_MODEL_FILENAME = os.getenv("YOLO_MODEL_FILENAME", "yolov8n.onnx")
DETECTION_CONFIDENCE_THRESHOLD = float(os.getenv("DETECTION_CONFIDENCE_THRESHOLD", "0.25"))
DETECTION_IOU_THRESHOLD = float(os.getenv("DETECTION_IOU_THRESHOLD", "0.45"))
MAX_DETECTIONS = int(os.getenv("MAX_DETECTIONS", "3"))
CROP_PADDING_RATIO = float(os.getenv("CROP_PADDING_RATIO", "0.10"))