Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Web GUI #23

Merged
merged 8 commits into from
Jan 1, 2021
Merged

Web GUI #23

Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
90fce7c
runway setup
AK391 Dec 27, 2020
45b9188
cv2 oil
AK391 Dec 27, 2020
31ed70a
100 strokes pil
AK391 Dec 27, 2020
a988581
pillow
AK391 Dec 27, 2020
9f897ff
control number of strokes
AK391 Dec 27, 2020
bdecb1e
badge and desc
AK391 Dec 27, 2020
2709582
fix strokes
AK391 Dec 27, 2020
d080d6f
set min 100
AK391 Dec 28, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
control number of strokes
  • Loading branch information
@AK391
AK391 committed Dec 27, 2020
commit 9f897ff56343fe8f79773463084014dc09e1c1f7
5 changes: 3 additions & 2 deletions runway_model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@
args.canvas_color = 'black' # [black, white]
args.canvas_size = 512 # size of the canvas for stroke rendering'
args.keep_aspect_ratio = False # whether to keep input aspect ratio when saving outputs
args.max_m_strokes = 100 # max number of strokes
args.max_divide = 5 # divide an image up-to max_divide x max_divide patches
args.beta_L1 = 1.0 # weight for L1 loss
args.with_ot_loss = False # set True for imporving the convergence by using optimal transportation loss, but will slow-down the speed
Expand Down Expand Up @@ -90,12 +89,14 @@ def optimize_x(pt):

return final_rendered_image

@runway.command('translate', inputs={'source_imgs': runway.image(description='input image to be translated'),}, outputs={'image': runway.image(description='output image containing the translated result')})
@runway.command('translate', inputs={'source_imgs': runway.image(description='input image to be translated'), 'amount': runway.number(min=0, max=700, default=100)
}, outputs={'image': runway.image(description='output image containing the translated result')})
def translate(learn, inputs):
os.makedirs('images', exist_ok=True)
inputs['source_imgs'].save('images/temp.jpg')
paths = os.path.join('images','temp.jpg')
args.img_path = paths
args.max_m_strokes = inputs['amount']
pt = ProgressivePainter(args=args)
final_rendered_image = optimize_x(pt)
formatted = (final_rendered_image * 255 / np.max(final_rendered_image)).astype('uint8')
Expand Down