MAC下deeplab_V3+修改“local_test_mobilenetv2.sh”来测试自己的数据
程序员文章站
2024-01-16 09:08:28
...
上一篇博客“MAC下deeplab_V3+测试自带例程“local_test_mobilenetv2.sh””已经测试了自带例程,数据还算可以。这一节用这个shell来训练下自己的数据集。
-
首先复制datasets下的文件夹“pascal_voc_seg”到当前目录,改名为“bladder_voc_seg”,然后将自己的tfrecord数据复制到对应位置,图片数据也需要进行修改。
其中“train_on_trainval_set_mobilenetv2”下的文件夹都可以删除,shell会重新生成。
“init_models”可以全部删除,因为我的数据没有对应的预训练模型,都得从头开始训练。
“tftrecord”是复制过来的数据。
-
其次,复制“local_test_mobilenetv2.sh”到当前目录,并修改为“chuntest_mobilenetv2.sh”。然后修改内容如下:
#!/bin/bash
# Exit immediately if a command exits with a non-zero status.Segmentation
set -e
# Move one-level up to tensorflow/models/resea
上一篇: 【图像处理与分析】图像分割方法总结
下一篇: HIT CSAPP LAB3