From 0b3dc585a0b02a3426b45f5cd474f81a31d5bd7b Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 16 Jan 2019 02:05:42 +0100
Subject: uprez ui

---
 app/relay/modules/pix2pixhd.js | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

(limited to 'app/relay')

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,
   }
 }
-- 
cgit v1.2.3-70-g09d2