/Crack-Semantic-Segmentation

Real time crack segmentation using PyTorch, OpenCV and ONNX runtime

Primary LanguagePython

Unet Semantic Segmentation for Cracks

Real time Crack Segmentation using PyTorch, OpenCV, ONNX runtime

Dependencies:

Pytorch

OpenCV

ONNX runtime

CUDA >= 9.0

Instructions:

1.Train model with your datatset and save model weights (.pt file) using unet_train.py on supervisely.ly

2.Convert model weights to ONNX format using pytorch_to_onnx.py

3.Obtain real time inference using crack_det_new.py

Crack segmentation model files can be downloaded by clicking this link

Results:

Graphs:

alt text

alt text