def main(argv=None): # Configurations config = Config() config.DATA_DIR = [ './data/SiW_M_Makeup_Ob_Binary_Files', './data/SiW_M_Mask_Silicone_Binary_Files', './data/SiW_M_Makeup_Co_Binary_Files', './data/SiW_M_Mask_Paper_Binary_Files', './data/SiW_M_Makeup_Im_Binary_Files', './data/SiW_M_Mask_Mann_Binary_Files', './data/SiW_M_Replay_Binary_Files', './data/SiW_M_Partial_Cut_Binary_Files', './data/SiW_M_Mask_Half_Binary_Files', './data/SiW_M_Partial_Funnyeye_Binary_Files', './data/SiW_M_Partial_Paperglass_Binary_Files', './data/SiW_M_Mask_Trans_Binary_Files', './data/SiW_M_Paper_Binary_Files', './data/SiW_M_Live_Binary_Files', './data/SiW_M_Live_Test_Binary_Files' ] config.DATA_DIR_LIVE = ['./data/SiW_M_Live_Binary_Files'] config.DATA_DIR_TEST = ['./data/SiW_M_Live_Test_Binary_Files'] config.LOG_DIR = './logs/model' config.MODE = 'training' # config.MODE = 'testing' config.STEPS_PER_EPOCH_VAL = 180 config.display() # Get images and labels. # dataset_train = Dataset(config, 'train') # Build a Graph model = Model(config) # Train the model model.compile() model.train()
def main(argv=None): # Configurations config = Config() config.DATA_DIR = ['/data/'] config.LOG_DIR = './log/model' config.MODE = 'training' config.STEPS_PER_EPOCH_VAL = 180 config.display() # Get images and labels. dataset_train = Dataset(config, 'train') # Build a Graph model = Model(config) # Train the model model.compile() model.train(dataset_train, None)
from model.config import Config from model.model import Model if __name__ == "__main__": # Configurations config = Config() # Train # Epoch-1 learning rate = 0.0003 config.DATA_DIR = ["/home/umit/xDataset/deepFake-dat/Train_Fake_Much_1", "/home/umit/xDataset/deepFake-dat/Train_Live_Much_1", "/home/umit/xDataset/deepFake-dat/Train_Fake_Much_2", "/home/umit/xDataset/deepFake-dat/Train_Live_Much_2", "/home/umit/xDataset/deepFake-dat/Train_Fake_Much_3", "/home/umit/xDataset/deepFake-dat/Train_Live_Much_3", "/home/umit/xDataset/deepFake-dat/Train_Fake_Much_4", "/home/umit/xDataset/deepFake-dat/Train_Live_Much_4", "/home/umit/xDataset/deepFake-dat/Train_Fake_Much_5", "/home/umit/xDataset/deepFake-dat/Train_Live_Much_5", "/home/umit/xDataset/deepFake-dat/Train_Fake_Much_6" "/home/umit/xDataset/deepFake-dat/Train_Live_Much_6", "/home/umit/xDataset/deepFake-dat/Train_Fake_Much_7"] config.LOG_DIR = './log/model' config.MODE = 'training' config.STEPS_PER_EPOCH = 2000 config.MAX_EPOCH = 1000 config.LEARNING_RATE = 0.00001 #0.00005 #0.0001 #0.0005 #0.001 config.BATCH_SIZE = 20 # Validation