CorrosionDetection

KO | EN

gitlite — search

#python #java #python3 #arduino #golang #machine-learning #rust #html #flask #javascript #seismology #nodejs

CorrosionDetection

★ 16 Open GitHub ↗

A small project about corrosion detection.

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

## Corrosion Detection
A small project about corrosion detection (course project).
The pipeline is as follows:

![](https://raw.githubusercontent.com/wlsdzyzl/CorrosionDetection/main/images/pipeline.png)

For a given image, use `resnet` for image classification. If it's classified as rust, use `unet` to get the corrosion mask.

### Dataset

All the images from classification dataset are collected from internet (some of them are from [Oxford pet datasets](https://www.robots.ox.ac.uk/~vgg/data/pets/)). The small dataset for corrosion segmentation is annotated by myself ([Annotation tool](https://www.robots.ox.ac.uk/~vgg/software/via/via.html)). Because of the lack of large dataset, the performance might be limited.

![](https://raw.githubusercontent.com/wlsdzyzl/CorrosionDetection/main/images/annotation.jpg)

The dataset can be download [here](https://drive.google.com/file/d/1e_aZtxXpN4wBj65Xz0ch_4TLA2OAF7V-/view?usp=sharing).

### Selected results

- Classification

![](https://raw.githubusercontent.com/wlsdzyzl/CorrosionDetection/main/images/eval_resnet.png)

- Segmentation

![](https://raw.githubusercontent.com/wlsdzyzl/CorrosionDetection/main/images/eval_unet.png)

- Classification failure cases:

![](https://raw.githubusercontent.com/wlsdzyzl/CorrosionDetection/main/images/failure_case.png)

- Segmentation on unlabelled images:

![](https://raw.githubusercontent.com/wlsdzyzl/CorrosionDetection/main/images/test_results.png)

CorrosionDetection

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README