Spaces:
Sleeping
Sleeping
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"))
|