forked from HannahKniesel/WSCD
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Variables.py
50 lines (34 loc) · 1.62 KB
/
Variables.py
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import torch
IMG_SIZE = (224, 224)
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
NAMES = ['naked', 'budding', 'enveloped']
PATIENCE_EARLY_STOPPING = 5
MIN_ITERATIONS = 230000
N_VAL = 3
ADENO_TRAIN_DATA_PATH = "./Data/LargeScaleTEM/Adenovirus/train/Data/"
ADENO_TEST_DATA_PATH = "./Data/LargeScaleTEM/Adenovirus/test/Data/"
ADENO_VAL_DATA_PATH = "./Data/LargeScaleTEM/Adenovirus/validation/Data/"
NORO_TRAIN_DATA_PATH = "./Data/LargeScaleTEM/Norovirus/train/Data/"
NORO_TEST_DATA_PATH = "./Data/LargeScaleTEM/Norovirus/test/Data/"
NORO_VAL_DATA_PATH = "./Data/LargeScaleTEM/Norovirus/validation/Data/"
PAP_TRAIN_DATA_PATH = "./Data/LargeScaleTEM/Papilloma/train/Data/"
PAP_TEST_DATA_PATH = "./Data/LargeScaleTEM/Papilloma/test/Data/"
PAP_VAL_DATA_PATH = "./Data/LargeScaleTEM/Papilloma/validation/Data/"
ROT_TRAIN_DATA_PATH = "./Data/LargeScaleTEM/Rotavirus/train/Data/"
ROT_TEST_DATA_PATH = "./Data/LargeScaleTEM/Rotavirus/test/Data/"
ROT_VAL_DATA_PATH = "./Data/LargeScaleTEM/Rotavirus/validation/Data/"
HERPES_TEST_DATA_PATH = "./Data/Herpes/Crops/Test/Data/"
HERPES_VAL_DATA_PATH = "./Data/Herpes/Crops/Val/Data/"
HERPES_TRAIN_DATA_PATH = "./Data/Herpes/Crops/Train/Data/"
CLASSIFICATION = "bin"
LOCATION = "loc"
BOUNDINGBOX = "bb"
CLASSIFICATION_TIMINGS = "./Data/Herpes/Crops/TimingsBinary.pkl"
LOCATION_TIMINGS = "./Data/Herpes/Crops/TimingsLocation.pkl"
BOUNDINGBOX_TIMINGS = "./Data/Herpes/Crops/TimingsBoundingBox.pkl"
EM_PRETRAINED_WEIGHTS = "./pretrained_models/cem500k_mocov2_resnet50_200ep_pth.tar"
MAX_NUM_OBJ_HERPES = 10
HERPES_CAPSIDE_SIZE = 165
MAX_IOU = 0.01
OUTPUT_NEURONS = 1
BATCH_SIZE = 1# 32