Skip to content

Commit 55d6c99

Browse files
added code
1 parent c50ee49 commit 55d6c99

File tree

7 files changed

+3358
-1
lines changed

7 files changed

+3358
-1
lines changed

README.md

Lines changed: 58 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,58 @@
1-
# crack-detection
1+
# Cracks Detection
2+
Segmenting cracks over wall images
3+
4+
Crack Detection over walls or any surface is a challenging task as
5+
1. Cracks in Pixels/image are very few. (need Innovative Training Approach)
6+
2. This task is expected to be Fast, Light, and Portable.
7+
8+
This is a perfect project for someone who wants to learn end-to-end ML.(Training== >Deployment). I trained and benchmark a lightweight model that detects cracks in < 40 ms over i5-CPU ( on img dim (128,128) after ML optimization). Here are some images attached.
9+
10+
Applications: It can be widely used for object inspection and can be added to Industry Production Pipeline for objects like Chips, bakeries, Bottles, Glass, Pipes, Tiles, Medicines, etc. Unfortunately, most of these jobs are still manual in Industry. It can be easily replaced with EdgeAI and can shift a lot of company's money to inventory and marketing. What you need is a Pi-Camera + Platform hardware + 5 V charger and some time in Coding.
11+
12+
## Dataset:
13+
This crack segmentation dataset contains around 11,200 images which are merged from 12 available crack segmentation dataset. The name prefix of each image is assigned to the corresponding dataset that the image belong to. There're also images which contain no crack, which could be filtered out by the pattern "noncrack*"
14+
All the images in the dataset are resized to the size of (448, 448).
15+
16+
| -- Train
17+
| -- images
18+
| -- masks
19+
| -- Test
20+
| -- images
21+
| -- masks
22+
23+
24+
![Inference Image](img/sample.png)
25+
26+
27+
Ref: [Dataset](https://github.com/khanhha/crack_segmentation/blob/master/README.md)
28+
29+
30+
31+
32+
## Model:
33+
1. Trained Resnet U-net Model of input-dim: (224,224,3)
34+
35+
![Resnet U Net Model](img/model.png)
36+
37+
Ref: [UNET Model](https://github.com/matterport/Mask_RCNN)
38+
39+
2. Jupyter Notebook, `Kaggle-training.ipyb`: Model Trained and Inference in Kaggle GPU Notebook
40+
41+
42+
## Inference Image
43+
![Inference Image](img/inferenceImage.jpeg)
44+
45+
46+
47+
48+
## Requirements
49+
Python 3.7.8, TensorFlow 2.5, and other common packages listed in `requirements.txt`.
50+
51+
52+
## Installation
53+
1. Clone this repository
54+
2. Install dependencies
55+
``` pip3 install -r requirements.txt ```
56+
3. Run setup from the repository root directory
57+
``` python3 setup.py install ```
58+

img/inferenceImage.jpeg

68.4 KB
Loading

img/model.png

29.5 KB
Loading

img/result.jpeg

20.1 KB
Loading

img/sample.png

175 KB
Loading

0 commit comments

Comments
 (0)