|
import os |
|
|
|
|
|
class AppUtils(): |
|
_label_id_map = { |
|
0: { |
|
"Sky": 2, |
|
"Tree": 4, |
|
"Road": 52, |
|
}, |
|
1: { |
|
"Sky": 2, |
|
"Tree": 4, |
|
"Mountain": 16, |
|
"Water": 21, |
|
}, |
|
2: { |
|
"Sky": 2, |
|
"Mountain": 16, |
|
}, |
|
3: { |
|
"Sky": 2, |
|
"Ground": 13, |
|
"Mountain": 16, |
|
}, |
|
4: { |
|
"Sky": 2, |
|
"Mountain": 16, |
|
}, |
|
5: { |
|
"Sky": 2, |
|
"Mountain": 16, |
|
}, |
|
6: { |
|
"Sky": 2, |
|
"Tree": 4, |
|
"Mountain": 16, |
|
}, |
|
} |
|
_inst_id_map = { |
|
0: { |
|
"Sky": 362, |
|
"Tree": 604, |
|
"Cim": 2056, |
|
"Road": 6412, |
|
}, |
|
1: { |
|
"Sky": 362, |
|
"Tree": 604, |
|
"Mountain": 2056, |
|
"Water": 2661, |
|
}, |
|
2: { |
|
"Sky": 362, |
|
"Mountain": 2056, |
|
}, |
|
3: { |
|
"Sky": 362, |
|
"Ground": 1693, |
|
"Mountain": 2056, |
|
}, |
|
4: { |
|
"Sky": 362, |
|
"Mountain": 2056, |
|
}, |
|
5: { |
|
"Sky": 362, |
|
"Mountain": 2056, |
|
}, |
|
6: { |
|
"Sky": 362, |
|
"Tree": 604, |
|
"Mountain": 2056, |
|
}, |
|
} |
|
|
|
_save_paths = { |
|
"image": "gradio_files/samples/test_processed/images", |
|
"labels": "gradio_files/samples/test_processed/labels", |
|
"inst_map": "gradio_files/samples/test_processed/inst_map", |
|
"predefined_masks": "gradio_files/samples/test_processed/predefined_masks/type_0", |
|
"synthesized_image": "gradio_files/samples/synthesized_image" |
|
} |
|
|
|
@staticmethod |
|
def clear(): |
|
for save_path in AppUtils._save_paths.values(): |
|
AppUtils._create_folder(save_path) |
|
|
|
os.system("rm -rf gradio_files/samples/test_processed/images/*") |
|
os.system("rm -rf gradio_files/samples/test_processed/labels/*") |
|
os.system("rm -rf gradio_files/samples/test_processed/inst_map/*") |
|
os.system("rm -rf gradio_files/samples/test_processed/predefined_masks/type_0/*") |
|
|
|
@staticmethod |
|
def get_examples(): |
|
return [ |
|
[0, "gradio_files/samples/flickr-landscape/images/832-41253531765_83c1767ba9_o.png", "gradio_files/samples/flickr-landscape/colored/832-41253531765_83c1767ba9_o.png"], |
|
[1, "gradio_files/samples/flickr-landscape/images/3736-9818172074_156d4682f3_o.png", "gradio_files/samples/flickr-landscape/colored/3736-9818172074_156d4682f3_o.png"], |
|
[2, "gradio_files/samples/flickr-landscape/images/7343-9965972016_a822e52102_o.png", "gradio_files/samples/flickr-landscape/colored/7343-9965972016_a822e52102_o.png"], |
|
[3, "gradio_files/samples/flickr-landscape/images/7503-16108428460_622fcdb3ca_o.png", "gradio_files/samples/flickr-landscape/colored/7503-16108428460_622fcdb3ca_o.png"], |
|
[4, "gradio_files/samples/flickr-landscape/images/7921-47167099321_02f96ba4f6_o.png", "gradio_files/samples/flickr-landscape/colored/7921-47167099321_02f96ba4f6_o.png"], |
|
[5, "gradio_files/samples/flickr-landscape/images/8016-7167270731_b9843b1072_o.png", "gradio_files/samples/flickr-landscape/colored/8016-7167270731_b9843b1072_o.png"], |
|
[6, "gradio_files/samples/flickr-landscape/images/8042-7987076838_05973d5ee8_o.png", "gradio_files/samples/flickr-landscape/colored/8042-7987076838_05973d5ee8_o.png"], |
|
] |
|
|
|
@staticmethod |
|
def get_labels(input_id): |
|
return ["None"] + list(AppUtils._label_id_map[input_id].keys()) |
|
|
|
@staticmethod |
|
def get_inst_id(input_id, label): |
|
return AppUtils._inst_id_map[input_id][label] |
|
|
|
@staticmethod |
|
def get_label_id(input_id, label): |
|
return AppUtils._label_id_map[input_id][label] |
|
|
|
@staticmethod |
|
def _create_folder(directory): |
|
if not os.path.exists(directory): |
|
os.makedirs(directory) |
|
|
|
@staticmethod |
|
def copy_file(src_path, dest_path): |
|
os.system(f"cp {src_path} {dest_path}") |