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
|
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
|