使用mobilenet进行训练
Opened this issue · 0 comments
感谢您的分享!能否请教您点问题。
使用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
mobilenet base_network只返回一张特征图,
SSH_Tensorflow/libs/networks/mobilenet_v2.py
Line 111 in 590105b
请问我该如何修改?
@yangxue0827