import gradio as gr import torch from PIL import Image from torchvision import transforms import numpy as np from matplotlib import pyplot as plt from torch import nn from transformers import SegformerForSemanticSegmentation import sys import io import pdb ################### # Setup label names target_list = ['Crack', 'ACrack', 'Wetspot', 'Efflorescence', 'Rust', 'Rockpocket', 'Hollowareas', 'Cavity', 'Spalling', 'Graffiti', 'Weathering', 'Restformwork', 'ExposedRebars', 'Bearing', 'EJoint', 'Drainage', 'PEquipment', 'JTape', 'WConccor'] target_list_all = ["All"] + target_list classes, nclasses = target_list, len(target_list) label2id = dict(zip(classes, range(nclasses))) id2label = dict(zip(range(nclasses), classes)) ############ # Load model device = torch.device('cpu') segformer = SegformerForSemanticSegmentation.from_pretrained("nvidia/mit-b1", id2label=id2label, label2id=label2id) # SegModel class SegModel(nn.Module): def __init__(self, segformer): super(SegModel, self).__init__() self.segformer = segformer self.upsample = nn.Upsample(scale_factor=4, mode='nearest') def forward(self, x): return self.upsample(self.segformer(x).logits) model = SegModel(segformer) path = "runs/2023-08-31_rich-paper-12/best_model_cpu.pth" print(f"Load Segformer weights from {path}") #model = model.load_state_dict(torch.load(path, map_location=device)) model = torch.load(path) model.eval() ################## # Image preprocess ################## to_tensor = transforms.ToTensor() to_array = transforms.ToPILImage() resize = transforms.Resize((512, 512)) resize_small = transforms.Resize((369,369)) normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) def process_pil(img): img = to_tensor(img) img = resize(img) img = normalize(img) return img # the background of the image def resize_pil(img): img = to_tensor(img) img = resize_small(img) img = to_array(img) return img # combine the foreground (mask_all) and background (original image) to create one image def transparent(fg, bg, alpha_factor): foreground = np.array(fg) background = np.array(bg) background = Image.fromarray(bg) foreground = Image.fromarray(fg) new_alpha_factor = int(255*alpha_factor) foreground.putalpha(new_alpha_factor) background.paste(foreground, (0, 0), foreground) return background def show_img(all_imgs, dropdown, bg, alpha_factor): idx = target_list_all.index(dropdown) fg= all_imgs[idx]["name"] foreground = Image.open(fg) background = np.array(bg) background = Image.fromarray(bg) new_alpha_factor = int(255*alpha_factor) foreground.putalpha(new_alpha_factor) background.paste(foreground, (0, 0), foreground) return background ########### # Inference def inference(img, alpha_factor): background = resize_pil(img) img = process_pil(img) mask = model(img.unsqueeze(0)) # we need a batch, hence we introduce an extra dimenation at position 0 (unsqueeze) mask = mask[0] # Get probability values (logits to probs) mask_probs = torch.sigmoid(mask) mask_probs = mask_probs.detach().numpy() mask_probs.shape # Make binary mask THRESHOLD = 0.5 mask_preds = mask_probs > THRESHOLD # All combined mask_all = mask_preds.sum(axis=0) mask_all = np.expand_dims(mask_all, axis=0) mask_all.shape # Concat all combined with normal preds mask_preds = np.concatenate((mask_all, mask_preds),axis=0) labs = ["ALL"] + target_list fig, axes = plt.subplots(5, 4, figsize = (10,10)) # save all mask_preds in all_mask all_masks = [] for i, ax in enumerate(axes.flat): label = labs[i] all_masks.append(mask_preds[i]) ax.imshow(mask_preds[i]) ax.set_title(label) plt.tight_layout() # plt to PIL img_buf = io.BytesIO() fig.savefig(img_buf, format='png') im = Image.open(img_buf) # Saved all masks combined with unvisible xaxis und yaxis and without a white # background. all_images = [] for i in range(len(all_masks)): plt.figure() fig = plt.imshow(all_masks[i]) plt.axis('off') fig.axes.get_xaxis().set_visible(False) fig.axes.get_yaxis().set_visible(False) img_buf = io.BytesIO() plt.savefig(img_buf, bbox_inches='tight', pad_inches = 0, format='png') all_images.append(Image.open(img_buf)) return im, all_images, background examples=[ ["assets/dacl10k_v2_validation_0026.jpg", "dacl10k_v2_validation_0026.jpg"], ["assets/dacl10k_v2_validation_0037.jpg", "dacl10k_v2_validation_0037.jpg"], ["assets/dacl10k_v2_validation_0053.jpg", "dacl10k_v2_validation_0053.jpg"], ["assets/dacl10k_v2_validation_0068.jpg", "dacl10k_v2_validation_0068.jpg"], ["assets/dacl10k_v2_validation_0153.jpg", "dacl10k_v2_validation_0153.jpg"], ["assets/dacl10k_v2_validation_0263.jpg", "dacl10k_v2_validation_0263.jpg"], ["assets/dacl10k_v2_validation_0336.jpg", "dacl10k_v2_validation_0336.jpg"], ["assets/dacl10k_v2_validation_0500.jpg", "dacl10k_v2_validation_0500.jpg"], ["assets/dacl10k_v2_validation_0549.jpg", "dacl10k_v2_validation_0549.jpg"], ["assets/dacl10k_v2_validation_0609.jpg", "dacl10k_v2_validation_0609.jpg"] ] title = "dacl-challenge @ WACV2024" description = """

dacl-challenge @ WACV2024

Twitter/X | WACV2024 | arXiv | Python Toolkit | voxel51.com | eval.ai | dacl.ai workshop page

πŸ“› The challenge uses the dacl10k dataset, which stands for damage classification 10k images and is a multi-label semantic segmentation dataset for 19 classes (13 damages and 6 objects) present on bridges.

πŸ† The dataset is used in the dacl-challenge associated with the "1st Workshop on Vision-Based Structural Inspections in Civil Engineering" at WACV2024.

Civil engineering structures such as power plants, sewers, and bridges form essential components of the public infrastructure. It is mandatory to keep these structures in a safe and operational state. In order to ensure this, they are frequently inspected where the current recognition and documentation of defects and building components is mostly carried out manually. A failure of individual structures results in enormous costs. For example, the economic costs caused by the closure of a bridge due to congestion is many times the cost of the bridge itself and its maintenance.

Recent advancements in hardware and software offer great potential for increasing the quality, traceability, and efficiency of the structural inspection process. In particular, methods from the field of computer vision play an important role. The new techniques support the inspection engineer at the building site, raising quality and efficiency of the inspection. There is a high demand worldwide for the automation of structural inspections in the areas of building construction, bridge construction, tunnel construction, sewage plants, and other critical infrastructures.

In the β€œ1st Workshop on Vision-Based Structural Inspections in Civil Engineering,” approaches utilizing computer vision for analyzing and assessing civil engineering structures will be explored. The workshop will provide a platform for experts from both the academic and application community. The core of the workshop is the β€œdacl-challenge,” which aims to find the best models for recognizing bridge defects and bridge components by means of semantic segmentation. The challenge is based on the β€œdacl10k” dataset, a novel, real-world, large-scale benchmark for multi-label semantic segmentation that distinguishes between 13 defect types and six building components. The workshop will take place at the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV) 2024.

Details:

Workflow:

""" article = "

Github Repo

" with gr.Blocks() as app: with gr.Row(): gr.Markdown(description) with gr.Row(): input_img = gr.inputs.Image(type="pil", label="Original Image") gr.Examples(examples=examples, inputs=[input_img]) with gr.Row(): img = gr.outputs.Image(type="pil", label="All Masks") transparent_img = gr.outputs.Image(type="pil", label="Transparent Image") with gr.Row(): dropdown = gr.Dropdown(choices=target_list_all, label="Select Label", value="All") slider = gr.Slider(minimum=0, maximum=1, value=0.4, label="Alpha Factor") all_masks = gr.Gallery(visible=False) background = gr.Image(visible=False) generate_mask_slider = gr.Button("1) Generate Masks") generate_mask_slider.click(inference, inputs=[input_img], outputs=[img, all_masks, background]) submit_transparent_img = gr.Button("2) Generate Transparent Mask (with Alpha Factor)") submit_transparent_img.click(show_img, inputs=[all_masks, dropdown, background, slider], outputs=[transparent_img]) app.launch()