Commit 74652c2b authored by lucasb-eyer's avatar lucasb-eyer
Browse files

Utils example computes number of labels.

Fixes #13 continuation.
parent 42402df8
...@@ -25,8 +25,10 @@ fn_output = sys.argv[3] ...@@ -25,8 +25,10 @@ fn_output = sys.argv[3]
### Read images and annotation ### ### Read images and annotation ###
################################## ##################################
img = cv2.imread(fn_im) img = cv2.imread(fn_im)
labels = relabel_sequential(cv2.imread(fn_anno, 0))[0].flatten() labels, _, _ = relabel_sequential(cv2.imread(fn_anno, 0))
M = 21 # 21 Classes to match the C++ example
# Compute the number of classes in the label image
M = len(set(labels.flat))
########################### ###########################
### Setup the CRF model ### ### Setup the CRF model ###
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment