Spaces:
Sleeping
Sleeping
File size: 323 Bytes
8f260ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import torch
import os
DEVICE = torch.device("cpu")
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
BACKEND_DIR = os.path.abspath(os.path.join(BASE_DIR, '..'))
MODEL_PATH = os.path.join(BACKEND_DIR, 'models', 'checkpoint')
FAST_SAM_IMGSZ = 1024
FAST_SAM_CONF = 0.5
FAST_SAM_IOU = 0.6
FAST_SAM_RETINA_MASKS = True
|