summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-08 11:12:57 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-08 11:12:57 +0200
commitb81e1010b6179023ff7b4606def27afc0b0f5660 (patch)
tree90f20e73784cc8f33f3760b65784c3667cc30019
parentceb0ce2d658b7ecf10a092832d9d06f622762b4a (diff)
else
-rw-r--r--app/relay/modules/pix2pix.js3
-rw-r--r--rpc/rpc.py2
2 files changed, 3 insertions, 2 deletions
diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js
index fc612a8..803a5f2 100644
--- a/app/relay/modules/pix2pix.js
+++ b/app/relay/modules/pix2pix.js
@@ -102,6 +102,7 @@ const live = {
script: 'live-mogrify.py',
params: (task) => {
console.log(task)
+ const opt = task.opt || {}
return [
'--dataroot', path.join(cwd, 'sequences', task.module, task.dataset),
'--start_img', path.join(cwd, 'sequences', task.module, task.dataset, 'frame_00001.png'),
@@ -118,7 +119,7 @@ const live = {
'--transition-min', 0.05,
'--how_many', 1000000, '--transition-period', 1000,
'--loadSize', 256, '--fineSize', 256,
- '--just-copy', '--poll_delay', task.opt.poll_delay || 0.09,
+ '--just-copy', '--poll_delay', opt.poll_delay || 0.09,
'--model', 'test',
'--which_model_netG', 'unet_256',
'--which_direction', 'AtoB',
diff --git a/rpc/rpc.py b/rpc/rpc.py
index 82db460..9b45f26 100644
--- a/rpc/rpc.py
+++ b/rpc/rpc.py
@@ -74,7 +74,7 @@ class CortexRPC(object):
frame = output.getvalue()
output.close()
self.send_frame(fn, meta, frame)
- else:
+ else:
im.save(output, format='JPEG')
frame = output.getvalue()
output.close()