SSD输出mAP时出现TypeError: Can not convert a tuple into a Tensor or Operation

SSD-Tensorflow的版本为,https://github.com/balancap/SSD-Tensorflow

引用GitHub上一个人的做法,https://github.com/balancap/SSD-Tensorflow/issues/154

训练完模型,想测试下其mAP的代码


DATASET_DIR=tfrecords 

EVAL_DIR=logs/ 

CHECKPOINT_PATH=./logs/VGG_VOC0712_SSD_300x300_ft_iter_120000.ckpt 

python3 eval_ssd_network.py \ 

    --eval_dir=${EVAL_DIR} \ 

	--dataset_dir=${DATASET_DIR} \ 

	--dataset_name=pascalvoc_2007 \ 

	--dataset_split_name=test \ 

	--model_name=ssd_300_vgg \ 

	--checkpoint_path=${CHECKPOINT_PATH} \ 

	--batch_size=1

出现错误
TypeError: Can not convert a tuple into a Tensor or Operation
解决方法为在eval_ssd_network.py文件中添加下面一个函数:


def flatten(x):

    result = []

    for el in x:

         if isinstance(el, tuple):

               result.extend(flatten(el))

         else:

               result.append(el)

    return result

然后修改eval_ssd_network.py中两行代码


将    eval_op=list(names_to_updates.values())

 

改为  eval_op=flatten(list(names_to_updates.values()))

 

注意:共有两行!

猜你喜欢

转载自blog.csdn.net/summer2day/article/details/86075784