Added testcase

This commit is contained in:
Vighnesh Birodkar
2014-06-19 02:50:24 +05:30
parent a6c9a5a2a7
commit abcb9cf2ef
5 changed files with 34 additions and 33 deletions
+2 -2
View File
@@ -27,9 +27,9 @@ class RAG(nx.Graph):
def rag_meancolor(img,labels):
img = util.img_as_ubyte(img)
if img.ndim == 3 :
if img.ndim == 4 :
return _construct.construct_rag_meancolor_3d(img,labels)
elif img.ndim == 2 :
elif img.ndim == 3 :
return _construct.construct_rag_meancolor_2d(img,labels)
else :
raise ValueError("Image dimension not supported")