2017-04-15 151 views
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_cythonim2col_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_cythonim2col_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無效?有什麼方法可以解決它嗎?

謝謝大家提前!

+0

這不會對您有所幫助,但是有一長串與cs231課程發佈類似問題的人員列表,並且沒有得到任何真正的解決方案。 – DavidW

回答

0

通過完全卸載Anaconda3並安裝Anaconda2解決了該問題。然後我創建了新的環境並重新安裝了所有需要的軟件包。現在不會出現此錯誤