Diff of /mains/main_fAnoGAN.py [000000] .. [978658]

Switch to side-by-side view

--- 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']))