--- a +++ b/mains/main_fAnoGAN.py @@ -0,0 +1,29 @@ +#!/usr/bin/env python +import tensorflow as tf + +from models.fanogan import fanogan +from trainers.fAnoGAN import fAnoGAN +from utils import Evaluation +from utils.default_config_setup import get_config, get_options, get_datasets, Dataset + +# reset default graph +tf.reset_default_graph() + +tf.reset_default_graph() +dataset = Dataset.BRAINWEB +options = get_options(batchsize=8, learningrate=0.001, numEpochs=1, zDim=128, outputWidth=128, outputHeight=128) +options['data']['dir'] = options["globals"][dataset.value] +datasetHC, datasetPC = get_datasets(options, dataset=dataset) +config = get_config(trainer=fAnoGAN, options=options, optimizer='ADAM', intermediateResolutions=[16, 16], dropout_rate=0.1, dataset=datasetHC) + +config.kappa = 1.0 +config.scale = 10.0 + +# Create an instance of the model and train it +model = fAnoGAN(tf.Session(), config, network=fanogan) + +# Train it +model.train(datasetHC) + +# Evaluate +Evaluation.evaluate(datasetPC, model, options, description=f"{type(datasetHC).__name__}-{options['threshold']}", epoch=str(options['train']['numEpochs']))