From d98322eaa8a2b9eddf7fd204e213afb722590015 Mon Sep 17 00:00:00 2001 From: Logan Bauman Date: Fri, 6 May 2022 23:37:06 -0400 Subject: hi --- main.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index a074f71d..e5104756 100644 --- a/main.py +++ b/main.py @@ -1,7 +1,7 @@ import os import sys import argparse -import cv2 + import tensorflow as tf from skimage import transform @@ -55,12 +55,10 @@ def main(): style_image = imread(ARGS.style) - cv2.imshow('hi1', style_image) - cv2.waitKey(0) + style_image = transform.resize(style_image, content_image.shape) - cv2.imshow('hi2', style_image) - cv2.waitKey(0) + my_model = YourModel(content_image=content_image, style_image=style_image) my_model.vgg16.build([1, 255, 255, 3]) my_model.vgg16.load_weights('vgg16_imagenet.h5', by_name=True) -- cgit v1.2.3-70-g09d2