我的程序正在读取256x256的图像,该图像由三个值(0,1,2)标记。我的程序应该计算每个标签的像素数。但是,每当我运行程序时,它都不计算标签1 and 2
。
from __future__ import division
import caffe
import numpy as np
from scipy.misc import imread
from PIL import Image
from collections import Counter
sample_img_path='/path/to/image/label-002-014.png'
im=imread(sample_img_path)
print im.shape
cl0=np.count_nonzero(im == 0) #0=background class
cl1=np.count_nonzero(im == 1) #1=obj1,
cl2=np.count_nonzero(im == 2) #2=obj2,
print cl0,cl1,cl2
输出
(256, 256)
64313 0 0
你能指导我吗?怎么解决?