File size: 118 Bytes
6a6ee7b
9104277
 
d2542a3
 
 
1
2
3
4
5
6
7
import os
import torch

FRAME_RATE = 15
SCORE_THRESHOLD = 0.4
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"