visualTranslation/infer.sh at main · Bhashini-IITJ/visualTranslation · 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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#!/bin/bash
m2m=false
hin_eng=false
de=false

while [ "$1" != "" ]; do
    case $1 in
    "-i")
        shift
        input_folder=$1
        ;;
    "-o")
        shift
        output_folder=$1
        ;;
    "-f")
        shift
        input_file=$1
        ;;
    "--M2M")
        m2m=true
        ;;
    "--hin_eng")
        hin_eng=true
        ;;
    "--de")
        de=true
        ;;
    esac
    shift
done

mkdir -p tmp

## paragraph detection
conda activate itv2_hf

if [ "$de" = true ]; then
    python exclude_key_words.py --file "$input_file"
    python detect_para.py
else
    cp "$input_file" tmp/i_s_info.json
    cp "$input_file" tmp/para_info.json
    python form_para_info.py
fi

## translation
if [ "$de" = true ]; then
    if [ "$hin_eng" = true ]; then
        python translate_de.py --m2m $m2m
    else
        python translate_de.py --eng_to_hin --m2m $m2m
    fi
    python form_word_crops.py
else
    if [ "$hin_eng" = true ]; then
        python translate.py --m2m $m2m
    else
        python translate.py --eng_to_hin --m2m $m2m
    fi
fi

## cropping i_s
conda deactivate
conda activate srnet_plus_2
python generate_crops.py --folder "$input_folder"

## modifying i_s
python modify_crops.py

## creating i_t
python generate_i_t.py

## scene text eraser
conda deactivate
conda activate scene_text_eraser
python make_masks.py --folder "$input_folder"
python scene_text_eraser.py --folder "$input_folder"

## generating modified images
python make_output_base.py --folder "$input_folder"

## generating bg
python make_bg.py

## infer srnet_plus_2
conda deactivate
conda activate srnet_plus_2
python generate_o_t.py

## blend the crops
python blend_o_t_bg.py

## generate the final output
python create_final_images.py --output_folder "$output_folder"
rm -r tmp