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_op=list(names_to_updates.values())

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

注意:共有两行!

重新输入命令,OK~

  • 4
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值