Reformated to be a proper package

eigen.pyx → pydensecrf/eigen.pyx

 # distutils: language = c++
-# distutils: sources = eigen_impl.cpp
-# distutils: include_dirs = densecrf/include
+# distutils: sources = pydensecrf/eigen_impl.cpp
+# distutils: include_dirs = pydensecrf/densecrf/include
 
 
 # [::1] means we want a C-contiguous array.

test.py → pydensecrf/test.py

@@ -19,9 +19,9 @@ Up = (U + 1) / (np.sum(U, axis=0) + 2)
 img = np.zeros((10,10,3), dtype=np.uint8)
 img[2:8,2:8,:] = 255
 
-# d.setUnaryEnergy(-np.log(Up))
-# d.setUnaryEnergy(PyConstUnary(-np.log(Up)))
+d.setUnaryEnergy(-np.log(Up))
+#d.setUnaryEnergy(PyConstUnary(-np.log(Up)))
 
+d.addPairwiseBilateral(2, 2, img, 3)
 # d.addPairwiseBilateral(2, 2, img, 3)
-# d.addPairwiseBilateral(2, 2, img, 3)
-# np.argmax(d.inference(10), axis=0).reshape(10,10)
+np.argmax(d.inference(10), axis=0).reshape(10,10)

utils.py → pydensecrf/utils.py

@@ -54,7 +54,7 @@ def create_pairwise_gaussian(sdims, shape):
 
     """
     # create mesh
-    cord_range = [range(s) for s in shape]
+    hcord_range = [range(s) for s in shape]
     mesh = np.array(np.meshgrid(*cord_range, indexing='ij'), dtype=np.float32)
 
     # scale mesh accordingly

setup.py

@@ -13,6 +13,7 @@ setup(
     author="Lucas Beyer",
     author_email="lucasb.eyer.be@gmail.com",
     url="http://github.com/lucasb-eyer/pydensecrf",
-    ext_modules=cythonize(['eigen.pyx', 'densecrf.pyx']),
+    ext_modules=cythonize(['pydensecrf/eigen.pyx', 'pydensecrf/densecrf.pyx']),
+    py_modules = ["pydensecrf/utils"]
 )