summaryrefslogtreecommitdiff
path: root/app/relay/modules/pix2pix.js
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-07 01:25:27 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-07 01:25:27 +0200
commit5bf3e93a4138efc8f44056806a97b4f9e0a2d506 (patch)
tree7ceae89e6caa77817f0f907fc1bee86441c4c543 /app/relay/modules/pix2pix.js
parent48922f830b4cdbffdb7e8b61ea869cd89328052f (diff)
scriptz
Diffstat (limited to 'app/relay/modules/pix2pix.js')
-rw-r--r--app/relay/modules/pix2pix.js9
1 files changed, 4 insertions, 5 deletions
diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js
index 72d5c7e..01044aa 100644
--- a/app/relay/modules/pix2pix.js
+++ b/app/relay/modules/pix2pix.js
@@ -10,7 +10,6 @@ const cwd = process.env.PIX2PIX_CWD || path.join(process.env.HOME, 'code/' + nam
- ffmpeg movie into frames
- unzip zip file into sequence
- list sequences
- -
*/
const fetch = {
@@ -21,8 +20,7 @@ const fetch = {
return [ task.module, task.opt.url ]
},
listen: (task, line, i) => {
- // here i need to bridge again... get the filename that comes back from youtube-dl
- // and tell the cortex that URL -> fn and add the filename!
+ // relay the new dataset name from youtube-dl or w/e
if ( line.match(/^created dataset: /) ) {
let filename = line.split(': ')[1].trim()
task.dataset = filename.split('.')[0]
@@ -93,10 +91,11 @@ const live = {
script: 'live-mogrify.py',
params: (task) => {
return [
- '--dataroot', task.module + '/sequences/' + task.dataset,
- '--start_img', task.module + '/sequences/' + task.dataset + '/frame_00001.png',
+ '--dataroot', '/sequences/' + task.module + '/' + task.dataset,
+ '--start_img', '/sequences/' + task.module + '/' + task.dataset + '/frame_00001.png',
'--experiment', task.checkpoint,
'--name', task.checkpoint,
+ '--module-name', task.module,
'--recursive', '--recursive-frac', 0.1,
'--sequence', '--sequence-frac', 0.3,
'--process-frac', 0.5,