27、工业制造中的实时缺陷检测与云分布式训练

工业制造中的实时缺陷检测与云分布式训练

1. SSD 模型训练

1.1 准备工作

现在我们已拥有 TFRecord 格式的合适输入集，可用于训练 SSD 模型。训练步骤如下：
1. 下载预训练的 SSD 模型用于迁移学习，可从 此链接 下载合适的模型。代码如下：

%%shell
mkdir pre-trained-model
cd pre-trained-model
wget http://download.tensorflow.org/models/object_detection/tf2/20200711/ssd_resnet50_v1_fpn_640x640_coco17_tpu-8.tar.gz
tar -xvf ssd_resnet50_v1_fpn_640x640_coco17_tpu-8.tar.gz

1.2 编辑配置文件

编辑 pipeline.config 文件，以下是需要编辑的部分：

model {
  ssd {
    num_classes: 6
    image_resizer {
      fixed_shape_resizer {
        height: 64