Skip to content
This repository has been archived by the owner on Sep 22, 2024. It is now read-only.

DZDL/CorrosionDetector

 
 

Repository files navigation

CorrosionDetector [Web + Colab]


This notebook uses Transfer Learning on VGG16 Convnet, for the purpose of identifying images with rust. Achieves 87% F1-score by training on images downloaded from internet. Can correctly identify reddish brown painted images as no-rust.
The second part contains the notebook used to localize Rust using Tensorflow Object Detection API

Run on FloydHub

Languages

  • Jupyter Notebook 99.6%
  • Other 0.4%