代码之家 › 专栏 › 技术社区 › rootkitchao

SSD-shufflenet-V2-FPN比Mobilenet V2慢

object-detection tensorflow

rootkitchao · 技术社区 · 6 年前

我在上面写了一些自定义代码 tensorflow- models/object_detection SSD-shufflenet-v2-FPN (基于shufflenet v2 1.0)和SSD-mobilenet-v2-FPN(基于mobilenet v2 1.0)。我在自己的数据集上训练他们。

当使用相同的输入时,SSD-shufflenet-v2-fpn的时间是SSD-mobilenet-v2-fpn的三倍(使用1080*1920输入,4*ARM Cortex-A72内核和Android 8.0,SSD-shufflenet-v2-fpn每张图像的成本为1200ms,SSD-mobilenet-v2-fpn仅为400ms)

我试着用一个 third-party basic network structure

在shufflenet v2的论文中,shufflenet v2 1.0比mobilenet v2 1.0快得多,无论是在GPU上还是在硬件上手臂。有有人试过这两个网络吗?

实现SSD修改版本的方法非常简单。完成shufflenet v2的代码后,修改 ssd_mobilenet_v1_fpn_feature_extractor.py

1 回复 | 直到 6 年前

Bagoly Sz. 6 年前

在 美孚网V2 http://openaccess.thecvf.com/content_cvpr_2018/papers/Sandler_MobileNetV2_Inverted_Residuals_CVPR_2018_paper.pdf ),比较了和架构,在节中 .

人们可能会注意到,ShuffleNet上的CPU时间是不可测量的,因为正如作者提到的,在进行实验时,在Tensorflow mobile框架中没有实现有效的组卷积和洗牌操作。这也可能是你的原因。

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

2 年前

Anshuman Sinha · 将tensforflow中的所有nan值转换为零

2 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

2 年前

rob me of my freedom · 在不使用Y_pred和Y_actual的tensorflow中编写自定义损失

2 年前

Chris · 如何捕获导入脚本功能的打印

2 年前

Jay Bhie Santos · 剩余GRU:ValueError:输入具有不兼容的形状。收到的形状(5)和(24)

2 年前

Danijel · C中TensorFlow模型文件的路径?

2 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

2 年前

Arya Anish · tensorflow安装中面临的问题

2 年前

so beautiful memory · 如何将我在本地训练的tensorflow模型正确部署到谷歌云人工智能平台?我部署了它,但没有图像返回

2 年前