用mobilenet模型跑tensorflow CNN的样例:image_retrain.py和label_image.py

系统是 ubuntu 16.04,tensorflow版本是1.6, cuDNN版本是7.0。git clone tensorflow后试着跑了一下image_retrain.py(以下简称retrain)和label_image.py(以下简称label)。retain训练新的mobilenet分类器成功,用label测试新分类器也成功。但是用label在tensorflow 1.4以下的环境中测试新分类器中出错。错误提示如下:


> InvalidArgumentError (see above for traceback): NodeDef mentions attr
> 'dilations' not in Op<name=Conv2D; signature=input:T, filter:T ->
> output:T; attr=T:type,allowed=[DT_HALF, DT_FLOAT];
> attr=strides:list(int); attr=use_cudnn_on_gpu:bool,default=true;
> attr=padding:string,allowed=["SAME", "
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值