File size: 162 Bytes
0077a91
 
 
 
 
1
2
3
4
5
6
import torch

MODEL_PATH = "xception_best_model.pth"
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
FRAME_SKIP = 5  # Every 5th frame