-
Notifications
You must be signed in to change notification settings - Fork 19
/
Copy pathsetup.py
22 lines (21 loc) · 817 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from distutils.core import setup, Extension
from Cython.Build import cythonize
from Cython.Distutils import build_ext
setup(
name = "salmetric",
version = "0.1",
description = "Measure F-measure and MAE scores for salient object detection.",
author = "Qibin (Andrew) Hou",
author_email = "[email protected]",
ext_modules = cythonize([
Extension(
"build.salmetric",
sources = ["./python/salmetric.pyx", "./src/sal_metric.cpp"],
include_dirs = ['./include', '/usr/local/include/', '/usr/local/include/opencv2/'],
libraries = ['opencv_core', 'opencv_highgui', 'opencv_imgproc', 'opencv_imgcodecs'],
library_dirs = ['/usr/local/lib/'],
language="c++",
),
]),
cmdclass = {'build_ext':build_ext},
)