在
https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/model_zoo.md
中有许多个模型。
先下一个试试效果如何:
http://download.tensorflow.org/models/deeplabv3_mnv2_dm05_pascal_trainaug_2018_10_01.tar.gz
由于
我的Tensoflow是1.0版。不能直接运行,先把模型编辑一下,去掉
“Conv2D”的属性“dilations”和“data_format”
以及“Cast”的属性“Truncate”
“ArgMax”的属性“output_type”,然后就可以运行了。
模型中的空洞卷积是用SpaceToBatchND 和 BatchToSpaceND,先要用C++实现这两个函数
1。SpaceToBatchND 函数:
//四维 pad1(左、上补边),pad2(右、下补边)
张量 * SpaceToBatchND(张量 &in,int block_shape,int pad1,int pad2)
{
张量* out= new张量(in.num);
out->value=new int[out->num];
// //批,高,宽,

本文介绍了使用C++实现语义分割模型Deeplab_v3的过程,针对Tensorflow 1.0版本进行了模型修改,删除了不兼容的属性。文章提到了实现关键步骤,包括SpaceToBatchND和BatchToSpaceND函数,并展示了模型的实际应用效果。
最低0.47元/天 解锁文章
3078

被折叠的 条评论
为什么被折叠?



