aboutsummaryrefslogtreecommitdiff
path: root/main.py
diff options
context:
space:
mode:
authorMichael Foiani <sotech117@michaels-mbp-3.devices.brown.edu>2022-05-04 17:37:55 -0400
committerMichael Foiani <sotech117@michaels-mbp-3.devices.brown.edu>2022-05-04 17:37:55 -0400
commitb83cd8eba837348bc194db8c1dd12e369602e4ac (patch)
tree843664e430f3f5d5f32d1e34de5a034ce5c934ae /main.py
parent8f9fbd734209dd21a7f9a070200ebd31de207a17 (diff)
got it to run, but it hangs
Diffstat (limited to 'main.py')
-rw-r--r--main.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/main.py b/main.py
index 063670b8..8363fcef 100644
--- a/main.py
+++ b/main.py
@@ -55,7 +55,9 @@ def main():
print('this is',ARGS.content)
content_image = imread(ARGS.content)
+ content_image = np.resize(content_image, (255, 255, 3))
style_image = imread(ARGS.style)
+ style_image = np.resize(style_image, (255, 255, 3))
my_model = YourModel(content_image=content_image, style_image=style_image)
train(my_model)