CompleteMe/inference_folder.sh at main · adobe-research/CompleteMe · GitHub

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
#!/bin/bash

# Model paths
MODEL_PATH="./checkpoints/completeme_pipeline"


# output directory
OUTPUT_DIR="./results"

# Resolution settings
WIDTH=512
HEIGHT=512

GUIDANCE_SCALE=7.5


# Benchmark settings
# for input image
IMAGE_PATH="./benchmark/input_all"
# for inpainting mask
MASK_PATH="./benchmark/inpainting_mask_all"
# for reference
REFERENCE_PATH="./benchmark/reference_all"
# for reference mask
REF_MASK_PATH="./benchmark/reference_mask_all"
# for prompt
PROMPT_PATH="./benchmark/all_prompt_pure"


GPU_ID=0

echo "Processing image: $IMAGE_PATH with mask: $MASK_PATH and reference: $REFERENCE_PATH"

# Execute the Python script with the provided arguments
CUDA_VISIBLE_DEVICES=$GPU_ID python inference_folder.py \
    --image_folder "$IMAGE_PATH" \
    --mask_folder "$MASK_PATH" \
    --reference_folder "$REFERENCE_PATH" \
    --model "$MODEL_PATH" \
    --width $WIDTH \
    --height $HEIGHT \
    --output_dir $OUTPUT_DIR \
    --reference_mask_folder "$REF_MASK_PATH" \
    --prompt_folder "$PROMPT_PATH" \
    --guidance_scale $GUIDANCE_SCALE


echo "All images processed."