All Projects → satyenrajpal → Concrete Crack Detection

satyenrajpal / Concrete Crack Detection

Crack detection for an autonomous UAV

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Concrete Crack Detection

Mrcnn Scene Recognition
MR-CNNs for Large-Scale Scene Recognition
Stars: ✭ 135 (-7.53%)
Mutual labels:  convolutional-neural-networks
Abnormal event detection
Abnormal Event Detection in Videos using SpatioTemporal AutoEncoder
Stars: ✭ 139 (-4.79%)
Mutual labels:  convolutional-neural-networks
Attribute Aware Attention
[ACM MM 2018] Attribute-Aware Attention Model for Fine-grained Representation Learning
Stars: ✭ 143 (-2.05%)
Mutual labels:  convolutional-neural-networks
Deep Steganography
Hiding Images within other images using Deep Learning
Stars: ✭ 136 (-6.85%)
Mutual labels:  convolutional-neural-networks
Image classifier
CNN image classifier implemented in Keras Notebook 🖼️.
Stars: ✭ 139 (-4.79%)
Mutual labels:  convolutional-neural-networks
Practical Machine Learning With Python
Master the essential skills needed to recognize and solve complex real-world problems with Machine Learning and Deep Learning by leveraging the highly popular Python Machine Learning Eco-system.
Stars: ✭ 1,868 (+1179.45%)
Mutual labels:  convolutional-neural-networks
Imagenet
Pytorch Imagenet Models Example + Transfer Learning (and fine-tuning)
Stars: ✭ 134 (-8.22%)
Mutual labels:  convolutional-neural-networks
Voicemap
Identifying people from small audio fragments
Stars: ✭ 144 (-1.37%)
Mutual labels:  convolutional-neural-networks
Deeplearning.ai
Stars: ✭ 139 (-4.79%)
Mutual labels:  convolutional-neural-networks
Cnn Quantization
Quantization of Convolutional Neural networks.
Stars: ✭ 141 (-3.42%)
Mutual labels:  convolutional-neural-networks
Easycnn
easy convolution neural network
Stars: ✭ 136 (-6.85%)
Mutual labels:  convolutional-neural-networks
Wsddn
Weakly Supervised Deep Detection Networks (CVPR 2016)
Stars: ✭ 138 (-5.48%)
Mutual labels:  convolutional-neural-networks
Bender
Easily craft fast Neural Networks on iOS! Use TensorFlow models. Metal under the hood.
Stars: ✭ 1,728 (+1083.56%)
Mutual labels:  convolutional-neural-networks
Reproduce Chexnet
Reproduce CheXNet
Stars: ✭ 136 (-6.85%)
Mutual labels:  convolutional-neural-networks
Shainet
SHAInet - a pure Crystal machine learning library
Stars: ✭ 143 (-2.05%)
Mutual labels:  convolutional-neural-networks
Chainer Cifar10
Various CNN models for CIFAR10 with Chainer
Stars: ✭ 134 (-8.22%)
Mutual labels:  convolutional-neural-networks
Hey Waldo
Labeled images of the Where's Waldo puzzle for use in classification and image recognition problems.
Stars: ✭ 138 (-5.48%)
Mutual labels:  convolutional-neural-networks
Binary Face Alignment
Real time face alignment
Stars: ✭ 145 (-0.68%)
Mutual labels:  convolutional-neural-networks
Livianet
This repository contains the code of LiviaNET, a 3D fully convolutional neural network that was employed in our work: "3D fully convolutional networks for subcortical segmentation in MRI: A large-scale study"
Stars: ✭ 143 (-2.05%)
Mutual labels:  convolutional-neural-networks
Image Caption Generator
[DEPRECATED] A Neural Network based generative model for captioning images using Tensorflow
Stars: ✭ 141 (-3.42%)
Mutual labels:  convolutional-neural-networks

Concrete-Crack-Detection

UPDATE(5/20/2019): That took longer than expected. I've finished porting over to PyTorch in the pytorch branch. I still have to obtain some decent results. Would require some hyperparameter tuning. Feel free to train your own models, shouldn't take longer than a couple of hours on CPU.

UPDATE(3/23/2019): Currently porting this to PyTorch as I'm out of touch with TensorFlow. Should be done in a week.

This repository contains the code for crack detection in concrete surfaces. It is a TensorFlow implementation of the paper by by Young-Jin Cha and Wooram Choi - "Deep Learning-Based Crack Damage Detection Using Convolutional Neural Networks".

CNN_Archi

The model acheived 85% accuracy on the validation set. A few results are shown below - results

MATLAB was used to prepare the data. Regions of Interest were sliced into smaller 128 x 128 pixel images and used for training -

roi

Dependencies required-

  • TensorFlow
  • OpenCV
  • Dataset -The data set can be downloaded from this link

The dataset.py file creates the training dataset class to be fed into the Convolutional Neural Network. This class automatically determines the number of classes by the number of folders in 'in_dir' (number of folders=number of classes)

NOTE: This script utilizes an already saved cache file. The cache file contains the filenames of the data. If the cache file doesn't exist, a new one created. If the data has been altered with, please delete the old cache file and run the script again. This applies for both Running.py and Train_CD.py. This was my first ever Deep Learning project hence, the naive approach. I'll streamline this once I get some time, or gladly accept a pull request!

The directory structure is assumed to be the following- (For example considering 3 classes)

  • in_dir/class1/ - Contains all the training images for class 1
    • test/ - Contains all the validation images for class 1
  • in_dir/class2/ - Contains all the training images for class 2
    • test/ - Contains all the validation images for class 2
  • in_dir/class3/ - Contains all the training images for class 3
    • test/ - Contains all the validation images for class 3

To train the network run the command with the following arguments:
python Train_CD.py

Argument Details Default
--in_dir path to in_dir folder cracky
--iter number of iterations to run the model for 1500
--save_folder Directory to save checkpoint CURRENT_DIR

After model has been trained, meta_files are saved into 'save_folder'. To test the model, run the command with the following arguments: python Running.py

Argument Details Default
--in_dir directory containing unlabeled test data cracky_test
--meta_file MetaFile path None (Will throw error if not given)
--CP_dir dir contatining checkpoint None (Will throw error if not given)
--save_dir dir to save output images CURRENT_DIR

TODO:

  • [ ] Streamline data loading; remove cache file system
  • [ ] Combine training and testing scripts into one
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].