forked from affinelayer/pix2pix-tensorflow
-
Notifications
You must be signed in to change notification settings - Fork 1
/
_run_cmds_p2p.bat
34 lines (19 loc) · 927 Bytes
/
_run_cmds_p2p.bat
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
# convert edge detect
py -3 edge_detect_batch.py -i <input_dir> -o <output_dir>
py -3 face_landmark_detect_batch.py -i <input_dir> -o <output_dir>
# merge abc images side by side
py -3 merge_sbs.py -a <img_a_dir> -b <img_b_dir> c <img_c_dir> -o <output_dir>
###############################
# run pix2pix in CPU Only Mode / CMD Only
SET CUDA_VISIBLE_DEVICES=""
echo %CUDA_VISIBLE_DEVICES%
#limit to 1 GPU
SET CUDA_VISIBLE_DEVICES=1
# pix2pix train mode
py -3 pix2pix.py -m train -i <training_input_imgs_dir> -o <checkpoint_output_path>
# pix2pix export mode
py -3 pix2pix.py -m export -c <checkpoint_dir> -o <export_model_dir>
# pix2pix inference batch output
py -3 inference_batch.py -m <trained_model_dir> -sbs <1> -i <input_images_path> -il <input_label_images_path> -o <output_images_path> -osbs <output_sbs_images_path>
# pix2pix inference webcam output
py -3 inference_webcam_tf.py -m <trained_model_dir>