CHanges to get Python3 working

This commit is contained in:
Vighnesh Birodkar
2014-06-19 02:50:25 +05:30
parent 5165ebe01e
commit 53318c44df
2 changed files with 6 additions and 5 deletions
+2 -2
View File
@@ -1,6 +1,6 @@
import rag
cimport numpy as cnp
import numpy as np
cimport numpy as cnp
import rag
def construct_rag_meancolor_3d(img, arr):
+4 -3
View File
@@ -1,6 +1,7 @@
import networkx as nx
import _construct
from skimage import util
from ._construct import construct_rag_meancolor_2d
from ._construct import construct_rag_meancolor_3d
class RAG(nx.Graph):
@@ -80,8 +81,8 @@ def rag_meancolor(img, labels):
img = util.img_as_ubyte(img)
if img.ndim == 4:
return _construct.construct_rag_meancolor_3d(img, labels)
return construct_rag_meancolor_3d(img, labels)
elif img.ndim == 3:
return _construct.construct_rag_meancolor_2d(img, labels)
return construct_rag_meancolor_2d(img, labels)
else:
raise ValueError("Image dimension not supported")