This repository contains a YOLO-based object detection model specifically trained to identify and localize cracks in various infrastructure surfaces, including concrete walls, floors, and brick facades.
The following graphics demonstrate the model's training progression and final performance metrics.
The model has been validated on diverse surfaces showing robust detection capabilities.
1from ultralytics import YOLO
2
3# Load the model
4model = YOLO('crack.pt')
5
6# Perform inference on an image
7results = model.predict('path/to/your/image.jpg', save=True, conf=0.5)
8
9# View results
10for result in results:
11 result.show()
The model was trained on a specialized dataset focused on structural cracks. It includes variations in lighting, surface textures, and crack sizes to ensure better generalization in real-world infrastructure inspections.