使用mobilenet进行训练

[junedgar]

感谢您的分享！能否请教您点问题。
使用mobilenetv2进行训练的时候，报错如下：
train() File "train.py", line 47, in train gtboxes_batch=gtboxes_and_label) File "../libs/networks/build_whole_network.py", line 211, in build_whole_detection_network feature_stride8, feature_stride16 = self.build_base_network(input_img_batch) File "/home/startdt/zhengjunjun/zhengpy27/local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 439, in __iter__ "Tensor objects are not iterable when eager execution is not " TypeError: Tensor objects are not iterable when eager execution is not enabled. To iterate over this tensor use tf.map_fn.
发现相对于使用resnet训练，返回的是C3,C4两个特征图，

SSH_Tensorflow/libs/networks/resnet.py

Line 122 in 590105b

return C3, C4

mobilenet base_network只返回一张特征图，

SSH_Tensorflow/libs/networks/mobilenet_v2.py

Line 111 in 590105b

return feature_to_crop

请问我该如何修改？
@yangxue0827