Skip to content

Commit

Permalink
Fix training command line arguments.
Browse files Browse the repository at this point in the history
I'm severely drunk and I accidentally pushed to master. Sorry!
  • Loading branch information
Ganonmaster committed Dec 22, 2017
1 parent c956cdf commit 66d59c3
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions lib/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ def __init__(self, description='default'):
decoder_B.load_weights(self.arguments.model_dir + '/decoder_B.h5')
except Exception as e:
print('Not loading existing training data.')
print(e)

self.process()

Expand All @@ -62,7 +63,7 @@ def parse_arguments(self, description):
parser.add_argument('-m', '--model-dir',
action=FullPaths,
dest="model_dir",
default="model",
default="models",
help="Model directory. This is where the training data will \
be stored. Defaults to 'model'")
parser.add_argument('-p', '--preview',
Expand Down Expand Up @@ -113,8 +114,8 @@ def show_sample(self, test_A, test_B):
cv2.imwrite('_sample.jpg', figure)

def process(self):
images_A = get_image_paths(self.arguments.images_A)
images_B = get_image_paths(self.arguments.images_B)
images_A = get_image_paths(self.arguments.input_A)
images_B = get_image_paths(self.arguments.input_B)
images_A = load_images(images_A) / 255.0
images_B = load_images(images_B) / 255.0

Expand Down

0 comments on commit 66d59c3

Please sign in to comment.