SSD-Tensorflow实验记录

最新推荐文章于 2021-02-23 14:08:16 发布

biyu_2020

最新推荐文章于 2021-02-23 14:08:16 发布

阅读量505

点赞数

分类专栏：深度学习文章标签： SSD Tensorflow

本文链接：https://blog.csdn.net/Biyu_csdn/article/details/81098448

版权

这篇博客记录了使用Tensorflow实现SSD目标检测的过程，包括数据格式转换，使用预训练模型测试，以及训练过程。作者从GitHub下载项目，将VOC2007数据集转换为TensorflowRecords格式，并提供了转换和测试的脚本。测试阶段，作者展示了如何运行测试脚本查看结果。在训练阶段，调整训练参数并创建训练脚本以开始模型训练。

摘要由CSDN通过智能技术生成

一数据格式转换

1.从github上下载项目并解压

2.下载VOC2007数据集,并放在如下目录:

3.格式转换

首先修改datasets / pascalvoc_to_tfrecords.py中83行, r ---> rb, 如下:

image_data = tf.gfile.FastGFile(filename, 'rb').read()

在VOC2007下建立convert_train.sh脚本,内容如下:

#!/bin/bash

DATASET_DIR=./image/train/
OUTPUT_DIR=./tfrecords/train
python ../tf_convert_data.py \
    --dataset_name=pascalvoc \
    --dataset_dir=${DATASET_DIR} \
    --output_name=voc_2007_train \
    --output_dir=${OUTPUT_DIR}

同样建立convert_test.sh脚本,内容如下:

#!/bin/bash

DATASET_DIR=./image/test/
OUTPUT_DIR=./tfrecords/test
python ../tf_convert_data.py \
    --dataset_name=pascalvoc \
    --dataset_dir=${DATASET_DIR} \
    --output_name=voc