lenny kogginz

1 files changed, 13 insertions, 6 deletions

diff --git a/app/relay/modules/morph.js b/app/relay/modules/morph.js
index 5523bb3..8625b70 100644
--- a/app/relay/modules/morph.js
+++ b/app/relay/modules/morph.js
@@ -5,20 +5,27 @@ const name = 'morph'
 const cwd = process.env.MORPH_CWD || path.join(process.cwd(), '..', 'pytorch-sepconv')
 
 const morph = {
-  type: 'pytorch',
-  script: 'test.pl',
+  type: 'pytorch_cpu',
+  script: 'run.py',
   params: (task) => {
+    console.log(task)
     let params = [
       '--dataset', task.opt.a,
-      '--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',
       '--endpoint', process.env.API_REMOTE + '/api/folder/' + task.folder_id + '/upload/',
     ]
+    if (task.opt.mode === 'image') {
+      //
+    } else {
+      params = params.concat([
+        '--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,
+      ])
+    }
     if (task.opt.mode === 'average') {
       params.push('--average-videos')
       params.push('True')