0
我在Ubuntu 16.04上遇到了python 2.7和Cython的問題。 我想從cs231n課程(卷積神經網絡)運行代碼。 但唯一的功能col2im_6d_cython不起作用。錯誤是:Python 2.7和Cython:全局名稱col2im_6d_cython未定義
NameError: global name 'col2im_6d_cython' is not defined
功能col2im_6d_cython在im2col_cython.pyx
定義:
def col2im_6d_cython(np.ndarray[DTYPE_t, ndim=6] cols, int N, int C, int H, int W,
int HH, int WW, int pad, int stride):
cdef np.ndarray x = np.empty((N, C, H, W), dtype=cols.dtype)
cdef int out_h = (H + 2 * pad - HH)/stride + 1
cdef int out_w = (W + 2 * pad - WW)/stride + 1
cdef np.ndarray[DTYPE_t, ndim=4] x_padded = np.zeros((N, C, H + 2 * pad, W + 2 * pad),
dtype=cols.dtype)
col2im_6d_cython_inner(cols, x_padded, N, C, H, W, HH, WW, out_h, out_w, pad, stride)
if pad > 0:
return x_padded[:, :, pad:-pad, pad:-pad]
return x_padded
其中col2im_6d_cython從調用的文件是fast_layers.py:
from cs231n.im2col_cython import col2im_cython, im2col_cython
from cs231n.im2col_cython import col2im_6d_cython
def conv_backward_strides(dout, cache):
x, w, b, conv_param, x_cols = cache
stride, pad = conv_param['stride'], conv_param['pad']
N, C, H, W = x.shape
F, _, HH, WW = w.shape
_, _, out_h, out_w = dout.shape
db = np.sum(dout, axis=(0, 2, 3))
dout_reshaped = dout.transpose(1, 0, 2, 3).reshape(F, -1)
dw = dout_reshaped.dot(x_cols.T).reshape(w.shape)
dx_cols = w.reshape(F, -1).T.dot(dout_reshaped)
dx_cols.shape = (C, HH, WW, N, out_h, out_w)
dx = col2im_6d_cython(dx_cols, N, C, H, W, HH, WW, pad, stride)
return dx, dw, db
c ol2im_cython和im2col_cython正常工作,但只有col2im_6d_cython不起作用。
在我看來,Cython安裝存在問題。我已經通過運行安裝它:蟒蛇setup.py build_ext --inplace
setup.py是:
from distutils.core import setup
from distutils.extension import Extension
from Cython.Build import cythonize
import numpy
extensions = [
Extension('im2col_cython', ['im2col_cython.pyx'],
include_dirs = [numpy.get_include()]
),
]
setup(
ext_modules = cythonize(extensions),
)
和安裝用Cython當我有一個警告:
Warning: Extension name 'im2col_cython' does not match fully qualified name 'cs231n.im2col_cython' of 'im2col_cython.pyx'
running build_ext
爲什麼只有col2im_6d_cython無效?有什麼方法可以解決它嗎?
謝謝大家提前!
這不會對您有所幫助,但是有一長串與cs231課程發佈類似問題的人員列表,並且沒有得到任何真正的解決方案。 – DavidW