File size: 927 Bytes
24f3fb6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import os
from dotenv import load_dotenv

load_dotenv()

# Qdrant
QDRANT_HOST = os.getenv("QDRANT_HOST")
QDRANT_API  = os.getenv("QDRANT_API")
QDRANT_COLLECTION = os.getenv("QDRANT_COLLECTION", "object_collection")

# S3
S3_BUCKET   = os.getenv("S3_BUCKET", "object-mem")
AWS_KEY     = os.getenv("S3_ACCESS_KEY")
AWS_SECRET  = os.getenv("S3_SECRET_KEY")
AWS_REGION  = os.getenv("S3_REGION", "us-east-1")

# Neo4j
NEO4J_URI   = os.getenv("NEO4J_URI")
NEO4J_USER  = os.getenv("NEO4J_USER")
NEO4J_PASS  = os.getenv("NEO4J_PASSWORD")

# SAM2
SAM_CHECKPT = os.getenv("SAM_CHECKPT", "./checkpoints/sam2.1_hiera_small.pt")
SAM_CFG     = os.getenv("SAM_CFG", "configs/sam2.1/sam2.1_hiera_s.yaml")

# GroundingDINO
DINO_CHECKPT = os.getenv("DINO_CHECKPT", "./checkpoints/groundingdino_swint_ogc.pth")
DINO_CFG     = os.getenv("DINO_CFG", "./configs/GroundingDINO_SwinT_OGC.py")

# CLIP
CLIP_MODEL  = os.getenv("CLIP_MODEL", "ViT-B/32")