mirror of https://github.com/da0c/DL_Course_SamU
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
13 lines
289 B
Python
13 lines
289 B
Python
4 years ago
|
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),)
|