2016-07-02 96 views
0

我必須根據H,S和V的某些最小值和最大值對Android中的圖像進行閾值處理。我按如下操作。Android中的顏色閾值(使用OpenCV)

Mat raw=new Mat(); 
Mat hsv=new Mat(); 
org.opencv.android.Utils.bitmapToMat(bitmap, raw); 
Imgproc.cvtColor(raw, hsv, Imgproc.COLOR_RGB2HSV); 
Imgproc.cvtColor(raw, raw, Imgproc.COLOR_RGB2GRAY); 
Mat thresh = new Mat(raw.size(), CvType.CV_8UC1); 
for(int x=0;x<raw.cols();x++) 
{ 
    for(int y=0;y<raw.rows();y++) 
    { 
     double[] data = hsv.get(y,x); 
     double H = data[0]; 
     double S = data[1]; 
     double V = data[2]; 
     if(H_MIN<=H && H<=H_MAX && S_MIN<=S && S<=S_MAX && V_MIN<=V && V<=V_MAX) { 
      data = new double[] {255}; 
      thresh.put(y,x, data); 
     } 
     else 
     { 
      data = new double[] {0}; 
      thresh.put(y,x, data); 
     } 
    } 
} 

這工作正常,但是對於大圖像是耗時的。有沒有更快,更有效的方法來做到這一點? (在OpenCV中是否有這樣的功能,可能呢?)

在Python中,我可以使用numpy來完成以下操作,它非常快。

img = cv2.imread(name,1) 
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) 
h = hsv[:,:,0] 
s = hsv[:,:,1] 
v = hsv[:,:,2] 

mask = np.zeros((img.shape[0],img.shape[1]), dtype='uint8') 
mask[((h >= minH) & (h <= maxH) & (s >= minS) & (s <= maxS) & (v >= minV) & (v <= maxV))] = 255 
+0

嘗試獲取/放置圖像數據每行而不是每個像素,這將加快執行。 – hariprasad

回答

0

在OpenCV中有一個庫函數來完成你正在做的事inRange(Mat src, Scalar lowerBound, Scalar upperBound, Mat dest)

Mat raw=new Mat(); 
Mat hsv=new Mat(); 
org.opencv.android.Utils.bitmapToMat(changedBitmap, raw); 
Imgproc.cvtColor(raw, hsv, Imgproc.COLOR_RGB2HSV); 
Imgproc.cvtColor(raw, raw, Imgproc.COLOR_RGB2GRAY); 
double maxH = 0; 
Mat thresh = new Mat(); 
Core.inRange(hsv, new Scalar(H_MIN, S_MIN, V_MIN), new Scalar(H_MAX, S_MAX, V_MAX), thresh);