Tensorflow对象检测API的Faster RCNN配置中的depth_multiplier和skip_last_stride

时间:2019-04-03 08:48:38

标签: tensorflow object-detection-api

我将在Tensorflow中使用FasterRCNN训练0.5 of Normal Feature Depth

FasterRCNN feature extractor有两个变量。

depth_multiplier and skip_last_stride.

我是否需要将depth_multiplier = 0.5和skip_last_stride = 50都设置?

这是唯一要设置的地方吗?

1 个答案:

答案 0 :(得分:0)

是的,我们只需要在faster_rcnn_mobilenet_v1_feature_extractor.py处更新self._depth_multiplier = 0.5。

skip_last_strideconv_depth_ratio_in_percentage保持不变。不确定目的。

然后需要在内部进行更新

def _extract_box_classifier_features(self, proposal_feature_maps, scope):

深度修改为

depth = lambda d: max(int(d * self._depth_multiplier, 16)

那一切都很好,可以使用0.5个深度通道进行FasterRCNN。