Skip to content

CCNYRoboticsLab/pytorch_Concrete_Inspection

Repository files navigation

Pytorch Version, here we implement VGG, Resnet, Alexnet, etc

Please go to: https://github.com/ccny-ros-pkg/concreteIn_inpection_VGGF/ for more details

System Requirements

Tested on Ubuntu 16.04, Pytorch 0.4.0, CUDA 8.0 0r 9.0

  • require python3.5
  • pytorch 0.4.0 or above

Concrete Inpection Dataset and Baseline @ CCNY Robotics Lab

Authors: Liang Yang, Bing LI, Wei LI, Zhaoming LIU, Guoyong YANG, Jizhong XIAO

CCNY Concrete Structure Spalling and Crack database (CSSC) that aims to assist the civil inspection of performing in an automatical approach. In the first generate of our work, we mainly focusing on dataset creation and prove the concepts of innovativity. We provide the first complete the detailed dataset for concrete spalling and crack defects witht the help from Civil Engineers, where we also show our sincere thanks to the under-graduate student at Hostos Community College for their effort on data labeling. For our experiments, we deliever an UAV to perform field data-collection and inspection, and also perform a 3D semantic metric resconstructiont.

If you find this could be helpful for your project, please cite the following related papers:

@INPROCEEDINGS{8575365, author={L. {Yang} and B. {Li} and W. {Li} and B. {Jiang} and J. {Xiao}}, booktitle={2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW)}, title={Semantic Metric 3D Reconstruction for Concrete Inspection}, year={2018}, volume={}, number={}, pages={1624-16248}, ISSN={2160-7516}, month={June},}

@inproceedings{yangi2018wall, title={Wall-climbing robot for visual and GPR inspection}, author={Yang{'\i}, Liang and Yang, Guoyong and Liu, Zhaoming and Chang, Yong and Jiang, Biao and Awad, Youssef and Xiao, Jizhong}, booktitle={2018 13th IEEE Conference on Industrial Electronics and Applications (ICIEA)}, pages={1004--1009}, year={2018}, organization={IEEE} }

[IROS 2017] Liang YANG, Bing LI, Wei LI, Zhaoming LIU, Guoyong YANG,Jizhong XIAO (2017). Deep Concrete Inspection Using Unmanned Aerial Vehicle Towards CSSC Database. 2017 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), One Page Abstract, PDF.

[ROBIO 2017] Liang YANG, Bing LI, Wei LI, Zhaoming LIU, Guoyong YANG,Jizhong XIAO (2017). A Robotic System Towards Concrete Structure Spalling And Crack Database. 2017 IEEE Int. Conf. on Robotics and Biomimetics (ROBIO 2017), Project.

The under going project

If you are interested in this project, please check the project link and our current 3D semantic pixel-level reconstruction project. Also, you can shoot Liang Yang an email any time for other authors.

How to implement

training

  1. First download the data from: https://github.com/ccny-ros-pkg/concreteIn_inpection_VGGF/

  2. Put the data in folder SPallData

  3. run script:

No need to download the data now, just run the following scipt

  • python3 main.py

test

  1. Change the model directory of the model

  2. run script:

  • python3 demo_test.py

How to change model

In file model.py, you can change any model you prefer:

  • #from models import resnet as resnet -from models import resnet as resnet -#from models import vgg as resnet -#from models.alexnet import * -def generate_model(opt):
  • model = resnet.resnet34(pretrained = True, num_classes=opt.n_classes)

define epoch, batch size, output folder

Please check file opts.py

Example result:

About

The Pytorch Implementation of Our Concrete Crack and Spall Detection

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages