SSD:TensorFlow中的单次多重检测器

时间:2024-12-02 15:06:43

SSD:TensorFlow中的单次多重检测器

SSD Notebook 包含 SSD TensorFlow 的最小示例。 很快,就检测出了两个主要步骤:在图像上运行SSD网络,并使用通用算法(top-k滤波和非最大抑制算法)对输出进行后处理。

以下是成功检测输出的两个示例:

SSD:TensorFlow中的单次多重检测器

为了运行这个 Notebook 你需要先解压 checkpoint files 在 ./checkpoint

unzip ssd_300_vgg.ckpt.zip

然后开始一个 jupyter Notebook

jupyter notebook notebooks/ssd_notebook.ipynb

项目地址:https://github.com/balancap/SSD-Tensorflow

更多 TensorFlow 教程:http://www.tensorflownews.com/