1 files changed, 21 insertions, 0 deletions
diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index f1323c7..7e08bee 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -251,6 +251,26 @@ const splice = {
   },
   after: 'train'
 }
+const uprez = {
+  type: 'pytorch',
+  script: 'uprez.py',
+  params: task => {
+    console.log(task)
+    /*
+     set folder based on whether the file is a result or recursive mp4
+    */
+    let folder
+    if (task.opt.is_result) {
+      folder = './results/' + task.dataset,
+    } else if (task.opt.is_recursive) {
+      folder = './recursive/' + task.dataset + '/' + task.opt.name + '/',
+    }
+    return [
+      '--folder', folder,
+      '--out_fn', task.dataset + "_" + Date.now(),
+    ]
+  }
+}
 
 function pad(num, size) {
   if (isNaN(parseInt(num))) return num;
@@ -271,5 +291,6 @@ export default {
     render,
     render_recursive,
     splice,
+    uprez,
   }
 }