Kaggle钢铁赛:基于PyTorch/UNet算法的钢材表面缺陷检测——(1)数据集分析和可视化
简介
Kaggle上有一个钢材表面缺陷检测的竞赛,是一个很好的将深度学习应用于传统材料检测的例子。对该赛题和解法的剖析,可以辅助理解深度学习的流程,及其应用于具体问题的套路。
这次解析分为两部分:
(1)第一部分,即本文,是一个预备性工作,即对该竞赛的数据集的分析和可视化,参考的是这个notebook——clear mask visualization and simple eda。感谢GoldFish的分享。
(2)第二部分,参见另一篇文章,即算法分析,参考的是Rishabh Agrahari的使用PyTorch框架及UNet算法的notebook。
其他参考文献:
* kaggle