morph backend

1 files changed, 44 insertions, 0 deletions

diff --git a/app/relay/modules/morph.js b/app/relay/modules/morph.js
new file mode 100644
index 0000000..e09f5ca
--- /dev/null
+++ b/app/relay/modules/morph.js
@@ -0,0 +1,44 @@
+import path from 'path'
+import moment from 'moment'
+
+const name = 'morph'
+const cwd = process.env.MORPH_CWD || path.join(process.cwd(), '..', 'pytorch-sepconv')
+
+const morph = {
+  type: 'pytorch',
+  script: 'test.pl',
+  params: (task) => {
+    let params = [
+      '--first', path.join(cwd, '..', task.opt.dataset_module, 'sequences', task.opt.a)
+      '--second', path.join(cwd, '..', task.opt.dataset_module, 'sequences', task.opt.b)
+      '--a-offset', task.opt.a_offset,
+      '--b-offset', task.opt.b_offset,
+      '--steps', task.opt.steps,
+      '--dilate', task.opt.dilate,
+      '--video-out', ['morph', task.opt.mode, task.opt.dataset, moment().format('%YYYY%MM%DD_%hh%mm')].join('_') + '.mp4'
+    ]
+    if (task.opt.mode === 'average') {
+      params.push('--average-videos')
+      params.push('True')
+    } else {
+      params.push('--mix-videos')
+      params.push('True')
+    }
+    return params
+  }
+
+  },
+  listen: (task, line, i) => {
+    if ( (parseInt(line) % 10) === 0) {
+      return { type: 'epoch', task, epoch: (i/10)|0 }
+    }
+    return null
+  }
+}
+
+export default {
+  name, cwd,
+  activities: {
+    morph,
+  }
+}