Skip to content

Latest commit

 

History

History
114 lines (93 loc) · 4.48 KB

README.md

File metadata and controls

114 lines (93 loc) · 4.48 KB

CT-CrackSeg

stars github arXiv IEEE Xplore ICIP 2023

This is the official pyTorch implementation of the paper A Convolutional-Transformer Network for Crack Segmentation with Boundary Awareness.

Network Architecture

Diagram of the proposed method

Visualization of the proposed Dilated Residual Block.

Diagram of the proposed moudle

Datasets

1. Crack500 dataset

The Crack500 dataset contains 500 images of size around 2000 × 1500 pixels taken by cell phones on main campus of Temple University. And each image was cropped into 16 non-overlapped image regions and only the region containing more than 1000 pixels of crack is kept. Through this way, the training data consists of 1,896 images, validation data contains 348 images, test data contains 1124 images. Download the Crack500 dataset from this link.

|-- Crack500
    |-- train
        |-- images
        |   |-- 20160222_081011_1_361.jpg
            ......
        |-- masks
        |   |-- 20160222_081011_1_361.png
            ......
    |-- test
        |-- images
        |   |-- 20160222_080933_361_1.jpg
            ......
        |-- masks
        |   |-- 20160222_080933_361_1.png
            ......
    |-- val
        |-- images
        |   |-- 20160222_080850_1_361.jpg
            ......
        |-- masks
        |   |-- 20160222_080850_1_361.png
            ......

2. DeepCrack dataset

The DeepCrack dataset is consist of 537 RGB color images with manually annotated segmentations. The images were divided into two main subsets: a training set with 300 images and a testing set with 237 ones. You can download the Deepcrack dataset from this link.

|-- DeepCrack
    |-- train
        |-- images
        |   |-- 7Q3A9060-1.jpg
            ......
        |-- masks
        |   |-- 7Q3A9060-1.png
            ......
    |-- test
        |-- images
        |   |-- 11125-1.jpg
            ......
        |-- masks
        |   |-- 11125-1.png
            ......

Training

python train_crack.py

It will run training for model and save the best weights for the validation set.

Testing

python evaluate_crack.py --output <path_to_dir>

It will represent performance measures and save related results in results folder.

Results

Visual comparison of the segmentation results:

Visual comparison of the segmentation results.

Evaluation metrics of competing methods:

Evaluation metrics of competing methods on Crack500 and DeepCrack.

Model weights

You can download the learned weights for each dataset in the following table.

Dataset Google Drive Baidu Yun
Crack500 link link
Deepcrack link link

Acknowledgment

CT-CrackSeg is based on following repos. We thanks for their great works:

Citation

If you find our code or paper useful, please consider citing:

@inproceedings{CTCrackSeg,
    title={A Convolutional-Transformer Network for Crack Segmentation with Boundary Awareness},
    author={Tao, Huaqi and Liu, Bingxi and Cui, Jinqiang and Zhang, Hong},
    booktitle={2023 IEEE International Conference on Image Processing (ICIP)},
    pages={86-90},
    year={2023},
    organization={IEEE}
}