PaddlePaddle · yzl19940819 · Dec 9, 2021
diff --git a/...ibrary/Fire_and_Smoke_Detection/README.md → ...le_Library/Fire&Smoke_Detection/README.md b/...ibrary/Fire_and_Smoke_Detection/README.md → ...le_Library/Fire&Smoke_Detection/README.md
diff --git a/...d_Smoke_Detection/accuracy_improvement.md → ...e&Smoke_Detection/accuracy_improvement.md b/...d_Smoke_Detection/accuracy_improvement.md → ...e&Smoke_Detection/accuracy_improvement.md
diff --git a/...tection/code/1.train_ppyolov2_imagenet.py → ...tection/code/1.train_ppyolov2_imagenet.py b/...tection/code/1.train_ppyolov2_imagenet.py → ...tection/code/1.train_ppyolov2_imagenet.py
diff --git a/...ion/code/2.train_ppyolov2_imagenet_aug.py → ...ion/code/2.train_ppyolov2_imagenet_aug.py b/...ion/code/2.train_ppyolov2_imagenet_aug.py → ...ion/code/2.train_ppyolov2_imagenet_aug.py
diff --git a/...e_Detection/code/3.train_ppyolov2_coco.py → ...e_Detection/code/3.train_ppyolov2_coco.py b/...e_Detection/code/3.train_ppyolov2_coco.py → ...e_Detection/code/3.train_ppyolov2_coco.py
diff --git a/...tection/code/4.train_ppyolov2_coco_aug.py → ...tection/code/4.train_ppyolov2_coco_aug.py b/...tection/code/4.train_ppyolov2_coco_aug.py → ...tection/code/4.train_ppyolov2_coco_aug.py
diff --git a/...n/code/5.train_ppyolov2_coco_aug_noSPP.py → ...n/code/5.train_ppyolov2_coco_aug_noSPP.py b/...n/code/5.train_ppyolov2_coco_aug_noSPP.py → ...n/code/5.train_ppyolov2_coco_aug_noSPP.py
diff --git a/...e/6.train_ppyolov2_r50_aug_COCO_addneg.py → ...e/6.train_ppyolov2_r50_aug_COCO_addneg.py b/...e/6.train_ppyolov2_r50_aug_COCO_addneg.py → ...e/6.train_ppyolov2_r50_aug_COCO_addneg.py
diff --git a/.../7.train_ppyolov2_r101_aug_COCO_addneg.py → .../7.train_ppyolov2_r101_aug_COCO_addneg.py b/.../7.train_ppyolov2_r101_aug_COCO_addneg.py → .../7.train_ppyolov2_r101_aug_COCO_addneg.py
diff --git a/...nd_Smoke_Detection/code/8.train_ppyolo.py → ...re&Smoke_Detection/code/8.train_ppyolo.py b/...nd_Smoke_Detection/code/8.train_ppyolo.py → ...re&Smoke_Detection/code/8.train_ppyolo.py
diff --git a/...oke_Detection/code/9.train_yolov3_coco.py → ...oke_Detection/code/9.train_yolov3_coco.py b/...oke_Detection/code/9.train_yolov3_coco.py → ...oke_Detection/code/9.train_yolov3_coco.py
diff --git a/...ry/Fire_and_Smoke_Detection/code/infer.py → ...ibrary/Fire&Smoke_Detection/code/infer.py b/...ry/Fire_and_Smoke_Detection/code/infer.py → ...ibrary/Fire&Smoke_Detection/code/infer.py
diff --git a/...y/Fire_and_Smoke_Detection/code/metric.py → ...brary/Fire&Smoke_Detection/code/metric.py b/...y/Fire_and_Smoke_Detection/code/metric.py → ...brary/Fire&Smoke_Detection/code/metric.py
diff --git a/.../Fire_and_Smoke_Detection/code/predict.py → ...rary/Fire&Smoke_Detection/code/predict.py b/.../Fire_and_Smoke_Detection/code/predict.py → ...rary/Fire&Smoke_Detection/code/predict.py
diff --git a/...d_Smoke_Detection/code/test_imgs/test.jpg → ...e&Smoke_Detection/code/test_imgs/test.jpg b/...d_Smoke_Detection/code/test_imgs/test.jpg → ...e&Smoke_Detection/code/test_imgs/test.jpg
diff --git a/...d_Smoke_Detection/docs/images/dataset.png → ...e&Smoke_Detection/docs/images/dataset.png b/...d_Smoke_Detection/docs/images/dataset.png → ...e&Smoke_Detection/docs/images/dataset.png
diff --git a/..._and_Smoke_Detection/docs/images/demo.png → ...Fire&Smoke_Detection/docs/images/demo.png b/..._and_Smoke_Detection/docs/images/demo.png → ...Fire&Smoke_Detection/docs/images/demo.png
diff --git a/...nd_Smoke_Detection/docs/images/deploy.png → ...re&Smoke_Detection/docs/images/deploy.png b/...nd_Smoke_Detection/docs/images/deploy.png → ...re&Smoke_Detection/docs/images/deploy.png
diff --git a/...nd_Smoke_Detection/docs/images/eval_1.png → ...re&Smoke_Detection/docs/images/eval_1.png b/...nd_Smoke_Detection/docs/images/eval_1.png → ...re&Smoke_Detection/docs/images/eval_1.png
diff --git a/...nd_Smoke_Detection/docs/images/eval_2.png → ...re&Smoke_Detection/docs/images/eval_2.png b/...nd_Smoke_Detection/docs/images/eval_2.png → ...re&Smoke_Detection/docs/images/eval_2.png
diff --git a/...nd_Smoke_Detection/docs/images/hard_1.png → ...re&Smoke_Detection/docs/images/hard_1.png b/...nd_Smoke_Detection/docs/images/hard_1.png → ...re&Smoke_Detection/docs/images/hard_1.png
diff --git a/...d_Smoke_Detection/docs/images/predict.jpg → ...e&Smoke_Detection/docs/images/predict.jpg b/...d_Smoke_Detection/docs/images/predict.jpg → ...e&Smoke_Detection/docs/images/predict.jpg
diff --git a/..._Detection/docs/images/predict_result.png → ..._Detection/docs/images/predict_result.png b/..._Detection/docs/images/predict_result.png → ..._Detection/docs/images/predict_result.png
diff --git a/...and_Smoke_Detection/docs/images/train.png → ...ire&Smoke_Detection/docs/images/train.png b/...and_Smoke_Detection/docs/images/train.png → ...ire&Smoke_Detection/docs/images/train.png
diff --git a/Paddle_Industry_Practice_Sample_Library/Hemtle Detection/images/1.png b/Paddle_Industry_Practice_Sample_Library/Hemtle Detection/images/1.png
diff --git a/...Sample_Library/Hemtle Detection/README.md → ...Sample_Library/Hemtle_Detection/README.md b/...Sample_Library/Hemtle Detection/README.md → ...Sample_Library/Hemtle_Detection/README.md
@@ -1,28 +1,19 @@
 # 安全帽检测
 
-> 基于Paddlex2.0API开发
 
 ## 1.项目说明
 
-在该项目中，主要向大家介绍如何使用目标检测来实现对安全帽的检测，涉及代码以及优化过程亦可用于其它目标检测任务等。
-
-在施工现场，对于来往人员，以及工作人员而言，安全问题至关重要。而安全帽更是保障施工现场在场人员安全的第一防线，因此需要对场地中的人员进行安全提醒。当人员未佩戴安全帽进入施工场所时，人为监管耗时耗力，而且不易实时监管，过程繁琐、消耗人力且实时性较差。针对上述问题，希望通过**视频监控->目标检测->智能督导**的方式智能、高效的完成此任务:
+在施工现场，现场往来和工作人员的安全问题极为重要，安全帽是保障现场人员安全的第一道防线。及时排查安全帽佩戴的规范性并给予提醒，可以大大降低施工安全隐患。但是人为监管耗时耗力、过程繁琐、消耗人力且实时性较差。
 
 <div align="center">
-<img src="./images/1.png"  width = "320" />
-<img src="./images/2.png"  width = "320" /></div>
+<img src="./images/1.png"  width = "600" />
+</div>
 
 
 **业务难点：**
 
-- **精度要求高** 由于涉及安全问题，需要精度非常高才能保证对施工场所人员的安全督导。需要专门针对此目标的检测算法进行优化，另外，还需要处理拍摄角度、光线不完全受控，安全帽显示不全、可能存在遮挡等情况。
-- **小目标检测** 由于实际使用过程中，人员里镜头较远，因此需要模型对小目标的检测有较低的漏检率。
-
-<div align="center">
-<img src="./images/3.jpg"  width = "1024" /></div>
-
-
-
+- **精度要求高** 由于涉及安全问题，对模型精度有较高要求，除了专门针对此目标的检测算法进行优化外，还需要处理拍摄角度、光线不完全受控，安全帽显示不全、可能存在遮挡等情况。
+- **小目标检测** 实际使用过程中，人员往往离镜头较远，因此需要模型对小目标的检测有较低的漏检率。
 
 ## 2.数据准备
 
@@ -202,4 +193,5 @@ paddlex --export_inference --model_dir=output/yolov3_darknet53/best_model --save
 <img src="images/14.png"  width = "1024" /></div>
 
 
-
+* 感谢[蔡敬辉](https://github.com/cjh3020889729)同学对本案例的贡献
+* 本项目数据来源于：[njvisionpower](https://github.com/njvisionpower/Safety-Helmet-Wearing-Dataset)整理收集的安全帽数据集，在此表达感谢。 
diff --git a/.../Hemtle Detection/accuracy_improvement.md → .../Hemtle_Detection/accuracy_improvement.md b/.../Hemtle Detection/accuracy_improvement.md → .../Hemtle_Detection/accuracy_improvement.md
diff --git a/...le_Library/Hemtle Detection/code/infer.py → ...le_Library/Hemtle_Detection/code/infer.py b/...le_Library/Hemtle Detection/code/infer.py → ...le_Library/Hemtle_Detection/code/infer.py
diff --git a/...le_Library/Hemtle Detection/code/train.py → ...le_Library/Hemtle_Detection/code/train.py b/...le_Library/Hemtle Detection/code/train.py → ...le_Library/Hemtle_Detection/code/train.py
diff --git a/Paddle_Industry_Practice_Sample_Library/Hemtle_Detection/images/1.png b/Paddle_Industry_Practice_Sample_Library/Hemtle_Detection/images/1.png
diff --git a/...le_Library/Hemtle Detection/images/10.png → ...le_Library/Hemtle_Detection/images/10.png b/...le_Library/Hemtle Detection/images/10.png → ...le_Library/Hemtle_Detection/images/10.png
diff --git a/...le_Library/Hemtle Detection/images/11.png → ...le_Library/Hemtle_Detection/images/11.png b/...le_Library/Hemtle Detection/images/11.png → ...le_Library/Hemtle_Detection/images/11.png
diff --git a/...le_Library/Hemtle Detection/images/12.png → ...le_Library/Hemtle_Detection/images/12.png b/...le_Library/Hemtle Detection/images/12.png → ...le_Library/Hemtle_Detection/images/12.png
diff --git a/...le_Library/Hemtle Detection/images/13.png → ...le_Library/Hemtle_Detection/images/13.png b/...le_Library/Hemtle Detection/images/13.png → ...le_Library/Hemtle_Detection/images/13.png
diff --git a/...le_Library/Hemtle Detection/images/14.png → ...le_Library/Hemtle_Detection/images/14.png b/...le_Library/Hemtle Detection/images/14.png → ...le_Library/Hemtle_Detection/images/14.png
diff --git a/...ple_Library/Hemtle Detection/images/2.png → ...ple_Library/Hemtle_Detection/images/2.png b/...ple_Library/Hemtle Detection/images/2.png → ...ple_Library/Hemtle_Detection/images/2.png
diff --git a/...ple_Library/Hemtle Detection/images/3.jpg → ...ple_Library/Hemtle_Detection/images/3.jpg b/...ple_Library/Hemtle Detection/images/3.jpg → ...ple_Library/Hemtle_Detection/images/3.jpg
diff --git a/...ple_Library/Hemtle Detection/images/4.png → ...ple_Library/Hemtle_Detection/images/4.png b/...ple_Library/Hemtle Detection/images/4.png → ...ple_Library/Hemtle_Detection/images/4.png
diff --git a/...ple_Library/Hemtle Detection/images/5.png → ...ple_Library/Hemtle_Detection/images/5.png b/...ple_Library/Hemtle Detection/images/5.png → ...ple_Library/Hemtle_Detection/images/5.png
diff --git a/...ple_Library/Hemtle Detection/images/6.png → ...ple_Library/Hemtle_Detection/images/6.png b/...ple_Library/Hemtle Detection/images/6.png → ...ple_Library/Hemtle_Detection/images/6.png
diff --git a/...ple_Library/Hemtle Detection/images/7.png → ...ple_Library/Hemtle_Detection/images/7.png b/...ple_Library/Hemtle Detection/images/7.png → ...ple_Library/Hemtle_Detection/images/7.png
diff --git a/...ple_Library/Hemtle Detection/images/8.png → ...ple_Library/Hemtle_Detection/images/8.png b/...ple_Library/Hemtle Detection/images/8.png → ...ple_Library/Hemtle_Detection/images/8.png
diff --git a/...ple_Library/Hemtle Detection/images/9.png → ...ple_Library/Hemtle_Detection/images/9.png b/...ple_Library/Hemtle Detection/images/9.png → ...ple_Library/Hemtle_Detection/images/9.png
diff --git a/Paddle_Industry_Practice_Sample_Library/README.md b/Paddle_Industry_Practice_Sample_Library/README.md
@@ -6,14 +6,14 @@
 
 | 行业     | 案例                                                         |
 | -------- | ------------------------------------------------------------ |
-| 智慧城市 | [火灾烟雾检测](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Fire_and_Smoke_Detection) |
+| 智慧城市 | [火灾烟雾检测](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Fire&Smoke_Detection) |
 | 智慧城市 | [人流量统计](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Pedestrian_Detection_and_Tracking) |
 | 智慧城市 | [车流量统计](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Vehicle_Detection_and_Tracking) |
-| 智慧城市 | [安全帽检测](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Hemtle%20Detection) |
+| 智慧城市 | [安全帽检测](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Hemtle_Detection) |
 | 智慧城市 | [电子快递单信息提取](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Waybill_Information_Extraction) |
-| 智能制造 | [工业指针型表计读数](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/meter_reader) |
-| 智能制造 | [钢材缺陷检测](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/paddlex_steel_defect_seg-master) |
-| 智能制造 | [钢筋计数](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/rebar_count) |
-| 智能制造 | [机械手抓取](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/robot_grab) |
+| 智能制造 | [工业指针型表计读数](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/meter_Reader) |
+| 智能制造 | [钢材缺陷检测](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Steel_Detection) |
+| 智能制造 | [钢筋计数](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Rebar_Count) |
+| 智能制造 | [机械手抓取](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Robot_Grab) |
 | 互联网   | [财报识别与关键字段抽取](https://github.com/PaddlePaddle/awesome-DeepLearning/tree/master/Paddle_Enterprise_CaseBook/Report_Recognition_and_Analysis) |
 
diff --git a/.../paddlex_steel_defect_seg-master/HRNet.md → ...e_Sample_Library/Steel_Detection/HRNet.md b/.../paddlex_steel_defect_seg-master/HRNet.md → ...e_Sample_Library/Steel_Detection/HRNet.md
@@ -1,3 +1,5 @@
+
+# 基于HRNet的模型优化
 在项目中，我们采用HRNET作为钢板缺陷分割的模型。具体代码请参考[train.py](./train.py)。
 
 运行如下代码开始训练模型(单机单卡)：
@@ -116,4 +118,7 @@ train_transforms = T.Compose([
 | -- | -- | -- | -- | -- | -- | -- | -- |
 |  HRNET | 150 | 800x128 | 16 | 0.01 | CrossEntropyLoss|RandomHorizontalFlip | 62.75%
 | HRNET  | 150 | 800x128 | 16 | 0.01 | CrossEntropyLoss+DiceLoss|RandomHorizontalFlip | 64.46%
-| HRNET  | 150 | 800x128 | 16 | 0.01 | CrossEntropyLoss+DiceLoss|RandomHorizontalFlip+RandomDistort | 64.46%
+| HRNET  | 150 | 800x128 | 16 | 0.01 | CrossEntropyLoss+DiceLoss|RandomCrop+RandomHorizontalFlip+RandomDistort+RandomBlur | 66.35%
+
+可以看到，通过添加数据增强又上升了1.89%。
+
diff --git a/...paddlex_steel_defect_seg-master/README.md → ..._Sample_Library/Steel_Detection/README.md b/...paddlex_steel_defect_seg-master/README.md → ..._Sample_Library/Steel_Detection/README.md
@@ -1,32 +1,21 @@
 # 钢材缺陷检测项目案例
-- [钢材缺陷检测项目案例](#钢材缺陷检测项目案例)
-  - [一、项目简介](#一项目简介)
-  - [二、 数据集分析](#二-数据集分析)
-    - [2.1 读取和分析文本数据](#21-读取和分析文本数据)
-    - [2.2 读取和分析图像数据](#22-读取和分析图像数据)
-    - [2.3 可视化数据集](#23-可视化数据集)
-  - [三、制作标准数据集](#三制作标准数据集)
-    - [3.1 PaddleX数据集格式说明](#31-paddlex数据集格式说明)
-    - [3.2 数据转换](#32-数据转换)
-    - [3.3 数据切分](#33-数据切分)
-  - [四、训练和验证](#四训练和验证)
-    - [4.1 基于HRNET进行优化](基于HRNet)
-    - [4.2 基于UNet进行优化](#42-模型预测)
-  - [五、部署](#五部署)
-    - [5.1 模型导出](#51-模型导出)
-    - [5.2 C#桌面部署](#52-c桌面部署)
+
 
 ## 一、项目简介
-本项目来源于Kaggle上一个[钢材表面缺陷检测竞赛](https://www.kaggle.com/c/severstal-steel-defect-detection/overview)，这也是一个非常好的将深度学习应用于传统工业材料检测的案例。本项目将使用百度飞桨[PaddleX](https://github.com/PaddlePaddle/PaddleX)深度学习算法套件进行开发，全流程剖析整个实现过程。
+作为生产制造过程中必不可少的一步，物体表面缺陷检测广泛应用于工业领域，如：3C、半导体、电子、汽车、化工、医药、轻工、军工等行业。而钢材作为基础的工业材料，在出厂之前的检测更加严格。
+
+针对上述场景，PaddleX提供可视化、自动化的AI质检方案，帮助企业提升生产效率，实现智能质检。
+
 
-钢材是现代最重要的建筑材料之一。刚材结构建筑能够抵抗自然和人为磨损，这使得这种材料在世界各地随处可见。在所有钢材加工环节中，平板钢的生产工艺特别精细。从加热、轧制，再到干燥和切割，需要几台机器协同操作，其中一个重要环节就是利用高清摄像头捕获的图像对加工环节中的钢材进行缺陷自动检测。Kaggle上举办的这场钢材缺陷竞赛希望各位参赛者利用机器学习来改进钢板表面缺陷检测算法精度，提高钢铁生产自动化程度。
 <div align="center">
-<img src="./images/steel_demo.png"  width = "300" />  
+<img src="./images/steel_demo.png"  width = "800" />  
 </div>
 
 ## 二、 数据集分析
 
-这个竞赛主要目的是定位和分类钢板表面缺陷，属于语义分割问题。对应的数据集可以从[Kaggle竞赛官网](https://www.kaggle.com/c/severstal-steel-defect-detection/data)下载，也可以从[ai studio](https://aistudio.baidu.com/aistudio/datasetdetail/12731?_=1634357183077)进行下载。下载下来后包含两个图像文件夹和两个标注文件：
+本项目来源于Kaggle上一个[钢材表面缺陷检测竞赛](https://www.kaggle.com/c/severstal-steel-defect-detection/overview)，这也是一个非常好的将深度学习应用于传统工业材料检测的案例。
+
+这个竞赛主要目的是定位和分类钢板表面缺陷，属于语义分割问题。对应的数据集可以从[Kaggle竞赛官网](https://www.kaggle.com/c/severstal-steel-defect-detection/data)下载，下载下来后包含两个图像文件夹和两个标注文件：
 <div align="center">
 <img src="./images/dataset.png"  width = "200" />  
 </div>
@@ -37,7 +26,7 @@
 * **train.csv**：该文件中存储训练图像的缺陷标注，有4类缺陷，ClassId = [1, 2, 3, 4]
 * **sample_submission.csv**：该文件是一个上传文件的样例，每个ImageID要有4行，每一行对应一类缺陷
 
-对于竞赛项目来说，拿到数据后我们首先要做的就是先对数据集进行分析。下面我们按照文本和图像逐步来分析这个数据集。
+拿到数据后我们首先要做的就是先对数据集进行分析。下面我们按照文本和图像逐步来分析这个数据集。
 
 ### 2.1 读取和分析文本数据
 假设下载下来的数据集放在名为severstal的文件夹中，首先读取csv文本数据
@@ -399,13 +388,8 @@ for idx in idx_class_triple:
 ## 三、制作标准数据集
 
 ### 3.1 PaddleX数据集格式说明
-从前面的数据分析中我们看到钢板图像的每个像素只属于1种缺陷类别（或者没缺陷），由于我们需要定位出钢板缺陷的精细区域，因此可以把这个任务看作是一个语义分割任务，即按照像素级别精度判断每个像素所属的缺陷类别。
-
-接下来我们就可以使用语义分割算法进行训练和验证。本项目采用百度飞桨PaddleX算法套件来快速训练。PaddleX是飞桨全流程开发工具，集飞桨核心框架、模型库、工具及组件等深度学习开发所需全部能力于一身，打通深度学习开发全流程。简单来说，使用PaddleX有两个明显的好处：
-* **快速验证算法模型**：无论是做工业项目还是参加类似Kaggle的竞赛，没有一个算法是通吃的，那么我们必然需要尝试很多不同的算法，这时候快速验证算法模型有效性就显得尤为重要。对于一个未知的深度学习任务来说，我们刚开始并不知道到底使用什么样的模型算法去处理才能得到一个比较好的基准值，这时候就需要有一个接口切换简单、实现快捷方便、算法功能齐全的算法库工具，能够让我们快速的去验证模型有效性，这里PaddleX就是这样一个比较好的工具，它包含了图像分类、分割、检测等常见的、实用的深度学习模型，而且接口进行了高度统一，我们可以使用它来快速的训练各个模型，并选取一个比较好的基准值。在这个基础上，我们再去进行算法优化。
-* **模型快速部署**：对于真实的工业任务来说，不光要训练得到一个高精度的算法模型，还需要考虑部署的便捷性和有效性。很多模型尽管精度很高，但是由于采用了非常嵌套的动态图模型编写方式，如果将其直接静态图导出或者转成onnx往往会遇到问题，那么这时候必然需要算法工程师手动去修改模型，这对于真实项目来说会极大的增加项目开发难度，降低项目开发效率。使用PaddleX的好处在于，在PaddleX中的模型基本都已经被验证过了，能够在多种平台上进行落地部署，比如C++推理、手机移动端推理等，同时还给出了详细的部署教程，因此，选择PaddleX进行真实深度学习产品开发或者打比赛都是一个不错的选择。
+从前面的数据分析中我们看到钢板图像的每个像素只属于1种缺陷类别（或者没缺陷），由于需要定位出钢板缺陷的精细区域，因此可以把这个任务看作是一个语义分割任务，即按照像素级别精度判断每个像素所属的缺陷类别。
 
-**PaddleX安装方法**参考[官网](https://github.com/PaddlePaddle/PaddleX)。本项目使用PaddleX v2.0.0版本进行实现。
 
 为了能够使用PaddleX进行语义分割训练，我们首先要对数据集格式进行转化。PaddleX语义分割算法[要求](https://github.com/PaddlePaddle/PaddleX/blob/release/2.0.0/docs/data/format/segmentation.md)数据集按照如下方式进行组织：
 
@@ -447,7 +431,6 @@ python prepare_dataset.py
 ```
 这个文件用来指明当前到底有哪些像素标签值（本项目共4种缺陷，对应标签为1、2、3、4，这里的0标签表示背景）
 
-转换完的数据集也可以直接从ai studio上进行[下载](https://aistudio.baidu.com/aistudio/datasetdetail/112904)。
 
 ### 3.3 数据切分
 在模型进行训练时，我们需要划分训练集，验证集和测试集，可直接使用paddlex命令将数据集随机划分。本项目将训练集、验证集和测试集按照8.5：1：0.5的比例划分。 PaddleX中提供了简单易用的API，方便用户直接使用进行数据划分。具体命令如下：
@@ -486,3 +469,6 @@ paddlex --export_inference --model_dir=output/hrnet/best_model --save_dir=output
 <div align="center">
 <img src="./images/csharp.png"  width = "800" />  
 </div>
+
+* 感谢[佟兴宇](https://github.com/txyugood)、[钱彬](https://github.com/qianbin1989228)两位同学对于本案例的贡献。
+* 示例图片引用说明：本项目中示例数据源于Kaggle[钢材表面缺陷检测竞赛](https://www.kaggle.com/c/severstal-steel-defect-detection/overview) 
diff --git a/...y/paddlex_steel_defect_seg-master/UNet.md → ...ce_Sample_Library/Steel_Detection/UNet.md b/...y/paddlex_steel_defect_seg-master/UNet.md → ...ce_Sample_Library/Steel_Detection/UNet.md
@@ -1,4 +1,4 @@
-# 精度优化
+# 基于UNet的精度优化
 
 本小节侧重展示在模型迭代过程中优化精度的思路，在本案例中，有些优化策略获得了精度收益，而有些没有。在其他质检场景中，可根据实际情况尝试这些优化策略。
 

diff --git a/..._steel_defect_seg-master/code/__init__.py → ..._Library/Steel_Detection/code/__init__.py b/..._steel_defect_seg-master/code/__init__.py → ..._Library/Steel_Detection/code/__init__.py
diff --git a/..._steel_defect_seg-master/code/calc_fps.py → ..._Library/Steel_Detection/code/calc_fps.py b/..._steel_defect_seg-master/code/calc_fps.py → ..._Library/Steel_Detection/code/calc_fps.py
diff --git a/...lex_steel_defect_seg-master/code/infer.py → ...ple_Library/Steel_Detection/code/infer.py b/...lex_steel_defect_seg-master/code/infer.py → ...ple_Library/Steel_Detection/code/infer.py
diff --git a/...defect_seg-master/code/prepare_dataset.py → ...y/Steel_Detection/code/prepare_dataset.py b/...defect_seg-master/code/prepare_dataset.py → ...y/Steel_Detection/code/prepare_dataset.py
diff --git a/...eel_defect_seg-master/code/train_hrnet.py → ...brary/Steel_Detection/code/train_hrnet.py b/...eel_defect_seg-master/code/train_hrnet.py → ...brary/Steel_Detection/code/train_hrnet.py
diff --git a/...el_defect_seg-master/images/4e8e7a28c.jpg → ...rary/Steel_Detection/images/4e8e7a28c.jpg b/...el_defect_seg-master/images/4e8e7a28c.jpg → ...rary/Steel_Detection/images/4e8e7a28c.jpg
diff --git a/...el_defect_seg-master/images/9d34d1ed3.jpg → ...rary/Steel_Detection/images/9d34d1ed3.jpg b/...el_defect_seg-master/images/9d34d1ed3.jpg → ...rary/Steel_Detection/images/9d34d1ed3.jpg
diff --git a/...ex_steel_defect_seg-master/images/all.png → ...le_Library/Steel_Detection/images/all.png b/...ex_steel_defect_seg-master/images/all.png → ...le_Library/Steel_Detection/images/all.png
diff --git a/Paddle_Industry_Practice_Sample_Library/Steel_Detection/images/csharp.png b/Paddle_Industry_Practice_Sample_Library/Steel_Detection/images/csharp.png
diff --git a/...teel_defect_seg-master/images/dataset.png → ...ibrary/Steel_Detection/images/dataset.png b/...teel_defect_seg-master/images/dataset.png → ...ibrary/Steel_Detection/images/dataset.png
diff --git a/...x_steel_defect_seg-master/images/fig1.png → ...e_Library/Steel_Detection/images/fig1.png b/...x_steel_defect_seg-master/images/fig1.png → ...e_Library/Steel_Detection/images/fig1.png
diff --git a/...x_steel_defect_seg-master/images/fig2.png → ...e_Library/Steel_Detection/images/fig2.png b/...x_steel_defect_seg-master/images/fig2.png → ...e_Library/Steel_Detection/images/fig2.png
diff --git a/...x_steel_defect_seg-master/images/fig3.png → ...e_Library/Steel_Detection/images/fig3.png b/...x_steel_defect_seg-master/images/fig3.png → ...e_Library/Steel_Detection/images/fig3.png
diff --git a/...ex_steel_defect_seg-master/images/gt1.jpg → ...le_Library/Steel_Detection/images/gt1.jpg b/...ex_steel_defect_seg-master/images/gt1.jpg → ...le_Library/Steel_Detection/images/gt1.jpg
diff --git a/...ex_steel_defect_seg-master/images/gt2.jpg → ...le_Library/Steel_Detection/images/gt2.jpg b/...ex_steel_defect_seg-master/images/gt2.jpg → ...le_Library/Steel_Detection/images/gt2.jpg
diff --git a/...defect_seg-master/images/gt_4e8e7a28c.jpg → ...y/Steel_Detection/images/gt_4e8e7a28c.jpg b/...defect_seg-master/images/gt_4e8e7a28c.jpg → ...y/Steel_Detection/images/gt_4e8e7a28c.jpg
diff --git a/...defect_seg-master/images/gt_9d34d1ed3.jpg → ...y/Steel_Detection/images/gt_9d34d1ed3.jpg b/...defect_seg-master/images/gt_9d34d1ed3.jpg → ...y/Steel_Detection/images/gt_9d34d1ed3.jpg
diff --git a/...x_steel_defect_seg-master/images/no_1.png → ...e_Library/Steel_Detection/images/no_1.png b/...x_steel_defect_seg-master/images/no_1.png → ...e_Library/Steel_Detection/images/no_1.png
diff --git a/...x_steel_defect_seg-master/images/no_2.png → ...e_Library/Steel_Detection/images/no_2.png b/...x_steel_defect_seg-master/images/no_2.png → ...e_Library/Steel_Detection/images/no_2.png
diff --git a/...x_steel_defect_seg-master/images/no_3.png → ...e_Library/Steel_Detection/images/no_3.png b/...x_steel_defect_seg-master/images/no_3.png → ...e_Library/Steel_Detection/images/no_3.png
diff --git a/...x_steel_defect_seg-master/images/no_4.png → ...e_Library/Steel_Detection/images/no_4.png b/...x_steel_defect_seg-master/images/no_4.png → ...e_Library/Steel_Detection/images/no_4.png
diff --git a/...x_steel_defect_seg-master/images/no_5.png → ...e_Library/Steel_Detection/images/no_5.png b/...x_steel_defect_seg-master/images/no_5.png → ...e_Library/Steel_Detection/images/no_5.png
diff --git a/...x_steel_defect_seg-master/images/pre1.jpg → ...e_Library/Steel_Detection/images/pre1.jpg b/...x_steel_defect_seg-master/images/pre1.jpg → ...e_Library/Steel_Detection/images/pre1.jpg
diff --git a/...x_steel_defect_seg-master/images/pre2.jpg → ...e_Library/Steel_Detection/images/pre2.jpg b/...x_steel_defect_seg-master/images/pre2.jpg → ...e_Library/Steel_Detection/images/pre2.jpg
diff --git a/...steel_defect_seg-master/images/result.png → ...Library/Steel_Detection/images/result.png b/...steel_defect_seg-master/images/result.png → ...Library/Steel_Detection/images/result.png
diff --git a/Paddle_Industry_Practice_Sample_Library/Steel_Detection/images/steel_demo.png b/Paddle_Industry_Practice_Sample_Library/Steel_Detection/images/steel_demo.png
diff --git a/...seg-master/images/visualize_4e8e7a28c.jpg → ..._Detection/images/visualize_4e8e7a28c.jpg b/...seg-master/images/visualize_4e8e7a28c.jpg → ..._Detection/images/visualize_4e8e7a28c.jpg
diff --git a/...seg-master/images/visualize_9d34d1ed3.jpg → ..._Detection/images/visualize_9d34d1ed3.jpg b/...seg-master/images/visualize_9d34d1ed3.jpg → ..._Detection/images/visualize_9d34d1ed3.jpg
diff --git a/...steel_defect_seg-master/images/yes1_1.png → ...Library/Steel_Detection/images/yes1_1.png b/...steel_defect_seg-master/images/yes1_1.png → ...Library/Steel_Detection/images/yes1_1.png
diff --git a/...steel_defect_seg-master/images/yes1_2.png → ...Library/Steel_Detection/images/yes1_2.png b/...steel_defect_seg-master/images/yes1_2.png → ...Library/Steel_Detection/images/yes1_2.png
diff --git a/...steel_defect_seg-master/images/yes1_3.png → ...Library/Steel_Detection/images/yes1_3.png b/...steel_defect_seg-master/images/yes1_3.png → ...Library/Steel_Detection/images/yes1_3.png
diff --git a/...steel_defect_seg-master/images/yes1_4.png → ...Library/Steel_Detection/images/yes1_4.png b/...steel_defect_seg-master/images/yes1_4.png → ...Library/Steel_Detection/images/yes1_4.png
diff --git a/...steel_defect_seg-master/images/yes1_5.png → ...Library/Steel_Detection/images/yes1_5.png b/...steel_defect_seg-master/images/yes1_5.png → ...Library/Steel_Detection/images/yes1_5.png
diff --git a/...steel_defect_seg-master/images/yes2_1.png → ...Library/Steel_Detection/images/yes2_1.png b/...steel_defect_seg-master/images/yes2_1.png → ...Library/Steel_Detection/images/yes2_1.png
diff --git a/...steel_defect_seg-master/images/yes2_2.png → ...Library/Steel_Detection/images/yes2_2.png b/...steel_defect_seg-master/images/yes2_2.png → ...Library/Steel_Detection/images/yes2_2.png
diff --git a/...steel_defect_seg-master/images/yes2_3.png → ...Library/Steel_Detection/images/yes2_3.png b/...steel_defect_seg-master/images/yes2_3.png → ...Library/Steel_Detection/images/yes2_3.png
diff --git a/...steel_defect_seg-master/images/yes2_4.png → ...Library/Steel_Detection/images/yes2_4.png b/...steel_defect_seg-master/images/yes2_4.png → ...Library/Steel_Detection/images/yes2_4.png
diff --git a/...steel_defect_seg-master/images/yes2_5.png → ...Library/Steel_Detection/images/yes2_5.png b/...steel_defect_seg-master/images/yes2_5.png → ...Library/Steel_Detection/images/yes2_5.png
diff --git a/...steel_defect_seg-master/images/yes3_1.png → ...Library/Steel_Detection/images/yes3_1.png b/...steel_defect_seg-master/images/yes3_1.png → ...Library/Steel_Detection/images/yes3_1.png
diff --git a/...steel_defect_seg-master/images/yes3_2.png → ...Library/Steel_Detection/images/yes3_2.png b/...steel_defect_seg-master/images/yes3_2.png → ...Library/Steel_Detection/images/yes3_2.png
diff --git a/...steel_defect_seg-master/images/yes3_3.png → ...Library/Steel_Detection/images/yes3_3.png b/...steel_defect_seg-master/images/yes3_3.png → ...Library/Steel_Detection/images/yes3_3.png
diff --git a/...steel_defect_seg-master/images/yes3_4.png → ...Library/Steel_Detection/images/yes3_4.png b/...steel_defect_seg-master/images/yes3_4.png → ...Library/Steel_Detection/images/yes3_4.png
diff --git a/...steel_defect_seg-master/images/yes3_5.png → ...Library/Steel_Detection/images/yes3_5.png b/...steel_defect_seg-master/images/yes3_5.png → ...Library/Steel_Detection/images/yes3_5.png
diff --git a/...steel_defect_seg-master/images/yes4_1.png → ...Library/Steel_Detection/images/yes4_1.png b/...steel_defect_seg-master/images/yes4_1.png → ...Library/Steel_Detection/images/yes4_1.png
diff --git a/...steel_defect_seg-master/images/yes4_2.png → ...Library/Steel_Detection/images/yes4_2.png b/...steel_defect_seg-master/images/yes4_2.png → ...Library/Steel_Detection/images/yes4_2.png
diff --git a/...steel_defect_seg-master/images/yes4_3.png → ...Library/Steel_Detection/images/yes4_3.png b/...steel_defect_seg-master/images/yes4_3.png → ...Library/Steel_Detection/images/yes4_3.png
diff --git a/...steel_defect_seg-master/images/yes4_4.png → ...Library/Steel_Detection/images/yes4_4.png b/...steel_defect_seg-master/images/yes4_4.png → ...Library/Steel_Detection/images/yes4_4.png
diff --git a/...steel_defect_seg-master/images/yes4_5.png → ...Library/Steel_Detection/images/yes4_5.png b/...steel_defect_seg-master/images/yes4_5.png → ...Library/Steel_Detection/images/yes4_5.png
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/README.md b/Paddle_Industry_Practice_Sample_Library/meter_reader/README.md
@@ -374,7 +374,7 @@ mkdir projects
 cd projects
 git clone https://github.com/PaddlePaddle/PaddleX.git
 cd PaddleX
-git checkout release/2.0.0
+git checkout develop
 
 ```
 ### Step2: 下载PaddleX Manufature SDK
@@ -484,4 +484,4 @@ Meter 2: 6.21739101
 在分割模型可视化的预测结果保存在`PaddleX\examples\meter_reader\deploy\cpp\meter_reader\out\build\x64-Release\output_seg`，可以点击进行查看：
 | 表1可视化分割结果 | 表2可视化分割结果|
 | -- | -- |
-| ![](./images/20190822_168_06-30-17-09-33-217.jpg) | ![](20190822_168_06-30-17-09-33-213.jpg) |
+| ![](./images/20190822_168_06-30-17-09-33-217.jpg) | ![](./images/20190822_168_06-30-17-09-33-213.jpg) |
diff --git a/...y_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/20190822_105.jpg b/...y_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/20190822_105.jpg
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/CMakeLists.txt b/Paddle_Industry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/CMakeLists.txt
diff --git a/..._Industry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/CMakeSettings.json b/..._Industry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/CMakeSettings.json
diff --git a/...ustry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/include/meter_config.h b/...ustry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/include/meter_config.h
diff --git a/...Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/include/reader_postprocess.h b/...Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/include/reader_postprocess.h
diff --git a/..._Industry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/meter_pipeline.yml b/..._Industry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/meter_pipeline.yml
diff --git a/...le_Industry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/meter_reader.cpp b/...le_Industry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/meter_reader.cpp
diff --git a/...ndustry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/src/meter_config.cpp b/...ndustry_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/src/meter_config.cpp
diff --git a/...y_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/src/reader_postprocess.cpp b/...y_Practice_Sample_Library/meter_reader/deploy/cpp/meter_reader/src/reader_postprocess.cpp
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/20190822_168.jpg b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/20190822_168.jpg
diff --git a/...Practice_Sample_Library/meter_reader/images/20190822_168_06-30-17-09-33-213.jpg b/...Practice_Sample_Library/meter_reader/images/20190822_168_06-30-17-09-33-213.jpg
diff --git a/...Practice_Sample_Library/meter_reader/images/20190822_168_06-30-17-09-33-217.jpg b/...Practice_Sample_Library/meter_reader/images/20190822_168_06-30-17-09-33-217.jpg
diff --git a/...dustry_Practice_Sample_Library/meter_reader/images/MeterReader_Architecture.jpg b/...dustry_Practice_Sample_Library/meter_reader/images/MeterReader_Architecture.jpg
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/det_data.png b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/det_data.png
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/scene_introduce.jpg b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/scene_introduce.jpg
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/seg_data.png b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/seg_data.png
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_1.png b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_1.png
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_2-1.png b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_2-1.png
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_2-2.png b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_2-2.png
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_3.png b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_3.png
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_4.png b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step5_4.png
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step6_det.png b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step6_det.png
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step6_seg.png b/Paddle_Industry_Practice_Sample_Library/meter_reader/images/step6_seg.png
diff --git a/...ndustry_Practice_Sample_Library/meter_reader/images/visualize_1624716128584.jpg b/...ndustry_Practice_Sample_Library/meter_reader/images/visualize_1624716128584.jpg
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/reader_infer.py b/Paddle_Industry_Practice_Sample_Library/meter_reader/reader_infer.py
@@ -263,7 +263,7 @@ def erode(self, seg_results, erode_kernel):
         eroded_results = seg_results
         for i in range(len(seg_results)):
             eroded_results[i]['label_map'] = cv2.erode(
-                seg_results[i]['label_map'], kernel)
+                seg_results[i]['label_map'].astype(np.uint8), kernel)
         return eroded_results
 
     def circle_to_rectangle(self, seg_results):

diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/train_detection.py b/Paddle_Industry_Practice_Sample_Library/meter_reader/train_detection.py
diff --git a/Paddle_Industry_Practice_Sample_Library/meter_reader/train_segmentation.py b/Paddle_Industry_Practice_Sample_Library/meter_reader/train_segmentation.py
diff --git a/...ustry_Practice_Sample_Library/paddlex_steel_defect_seg-master/images/csharp.png b/...ustry_Practice_Sample_Library/paddlex_steel_defect_seg-master/images/csharp.png
diff --git a/...y_Practice_Sample_Library/paddlex_steel_defect_seg-master/images/steel_demo.png b/...y_Practice_Sample_Library/paddlex_steel_defect_seg-master/images/steel_demo.png