Hugging Face's logo

Spaces:
TencentARC
/
ColorFlow
Running on Zero

App Files Community
2
ColorFlow
File size: 1,150 Bytes 
77771e4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
set -ex
# models
RESULTS_DIR='./results/test/western2manga'

# dataset
CLASS='color2manga'
MODEL='cycle_ganstft'
DIRECTION='BtoA' # from domain A to domain B
PREPROCESS='none'
LOAD_SIZE=512 # scale images to this size
CROP_SIZE=1024 # then crop to this size
INPUT_NC=1  # number of channels in the input image
OUTPUT_NC=3  # number of channels in the input image
NGF=48
NEF=48
NDF=32
NZ=64

# misc
GPU_ID=0   # gpu id
NUM_TEST=30 # number of input images duirng test
NUM_SAMPLES=1 # number of samples per input images
NAME=${CLASS}_${MODEL}

# command
CUDA_VISIBLE_DEVICES=${GPU_ID} \
python3 ./test.py \
  --dataroot ./datasets/${CLASS} \
  --results_dir ${RESULTS_DIR} \
  --checkpoints_dir ./checkpoints/${CLASS}/ \
  --name ${NAME} \
  --model ${MODEL} \
  --direction ${DIRECTION} \
  --preprocess ${PREPROCESS} \
  --load_size ${LOAD_SIZE} \
  --crop_size ${CROP_SIZE} \
  --input_nc ${INPUT_NC} \
  --output_nc ${OUTPUT_NC} \
  --nz ${NZ} \
  --netE conv_256 \
  --num_test ${NUM_TEST} \
  --n_samples ${NUM_SAMPLES} \
  --upsample bilinear \
  --ngf ${NGF} \
  --nef ${NEF} \
  --ndf ${NDF} \
  --center_crop \
  --color2screen \
  --no_flip