当前位置:主页 > 软件编程 > Python代码 >

解决python 无法加载downsample模型的问题

时间:2021-05-06 09:43:21 | 栏目:Python代码 | 点击:

downsample 在最新版本里面修改了位置

from theano.tensor.single import downsample (旧版本)

上面以上的的import会有error raise:

from theano.tensor.signal import downsample

ImportError: cannot import name 'downsample'

找到from theano.tensor.single import downsample所在文件,如:

...\lib\site-packages\lasagne\layers\pool.py

把 from theano.tensor.signal import downsample注释掉,改为

from theano.tensor.signal.pool import pool_2d

代码中运用到downsample的地方也要改掉

# 子采样
 pooled_out = downsample.max_pool_2d(
  input=conv_out,
  ds=poolsize,
  ignore_border=True
 )

这里直接将downsample改为pool会出错,因为里面相应的函数有变化

查看pool.py的源码,发现downsample.max_pool_2d()与pool. pool_2d()函数功能相同,用ws代替ds

 # 子采样
pooled_out = pool.pool_2d(
  input=conv_out,
  ws=poolsize,
  ignore_border=True
)
 

您可能感兴趣的文章:

相关文章