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

时间:2022-09-23 11:53:52

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注释掉,改为

?
1
from theano.tensor.signal.pool import pool_2d

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

?
1
2
3
4
5
6
# 子采样
 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

?
1
2
3
4
5
6
# 子采样
pooled_out = pool.pool_2d(
  input=conv_out,
  ws=poolsize,
  ignore_border=True
)

以上这篇解决python 无法加载downsample模型的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/qq_37950540/article/details/83374923