1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
import path from 'path'
import fs from 'fs'
const name = 'biggan'
const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/cli/')
const env = {
LD_LIBRARY_PATH: '/usr/local/cuda/lib64:/home/lens/ffmpeg_build/lib:' + process.env.TENSORRT_LIB_PATH,
TFHUB_CACHE_DIR: '/home/lens/.tfhub',
LC_ALL: 'C.UTF-8',
LANG: 'C.UTF-8',
}
const invert = {
type: 'pytorch_biggan',
script: 'cli.py',
params: (task) => {
console.log(task)
return [
'biggan', 'extract_dense_vectors',
'--folder_id', task.opt.folder_id,
'--stochastic_clipping', '--label_clipping', '--clip_interval', 200
]
}
}
const live = {
type: 'pytorch_biggan',
script: 'cli.py',
params: (task) => {
console.log(task)
const opt = task.opt || {}
return ['biggan', 'live']
},
after: 'render',
}
const render = {
type: 'pytorch_biggan',
script: 'cli.py',
params: (task) => {
console.log(task)
return [
'cortex', 'upload_video',
]
}
}
export default {
name, cwd, env,
activities: {
invert, live, render,
}
}
|
