File size: 4,218 Bytes
087921f |
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 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 |
import os
# Utility class for the demo
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}") |