You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.2 KiB
52 lines
1.2 KiB
|
|
import os
|
|
from datetime import datetime
|
|
|
|
#CIFAR100 dataset path (python version)
|
|
#CIFAR100_PATH = '/nfs/private/cifar100/cifar-100-python'
|
|
|
|
#mean and std of cifar100 dataset
|
|
CIFAR100_TRAIN_MEAN = (0.5070751592371323, 0.48654887331495095, 0.4409178433670343)
|
|
CIFAR100_TRAIN_STD = (0.2673342858792401, 0.2564384629170883, 0.27615047132568404)
|
|
|
|
GLAUCOMA_TRAIN_MEAN = (0.5070751592371323, 0.48654887331495095, 0.4409178433670343)
|
|
GLAUCOMA_TRAIN_STD = (0.2673342858792401, 0.2564384629170883, 0.27615047132568404)
|
|
|
|
MASK_TRAIN_MEAN = (2.654204690220496/255)
|
|
MASK_TRAIN_STD = (21.46473779720519/255)
|
|
|
|
#CIFAR100_TEST_MEAN = (0.5088964127604166, 0.48739301317401956, 0.44194221124387256)
|
|
#CIFAR100_TEST_STD = (0.2682515741720801, 0.2573637364478126, 0.2770957707973042)
|
|
|
|
#directory to save weights file
|
|
CHECKPOINT_PATH = 'checkpoint'
|
|
|
|
#total training epoches
|
|
EPOCH = 30000
|
|
step_size = 10
|
|
i = 1
|
|
MILESTONES = []
|
|
while i * 5 <= EPOCH:
|
|
MILESTONES.append(i* step_size)
|
|
i += 1
|
|
|
|
#initial learning rate
|
|
#INIT_LR = 0.1
|
|
|
|
#time of we run the script
|
|
TIME_NOW = datetime.now().isoformat()
|
|
|
|
#tensorboard log dir
|
|
LOG_DIR = 'runs'
|
|
|
|
#save weights file per SAVE_EPOCH epoch
|
|
SAVE_EPOCH = 10
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|