Vous êtes sur la page 1sur 5

Watershed.

java
package com.example.watershedsegmentation;
import java.util.ArrayList;
import java.util.List;
import
import
import
import
import
import
import
import
import
import
import

org.opencv.android.BaseLoaderCallback;
org.opencv.android.LoaderCallbackInterface;
org.opencv.android.OpenCVLoader;
org.opencv.android.Utils;
org.opencv.core.Core;
org.opencv.core.CvType;
org.opencv.core.Mat;
org.opencv.core.MatOfPoint;
org.opencv.core.Scalar;
org.opencv.highgui.Highgui;
org.opencv.imgproc.Imgproc;

import
import
import
import
import
import
import
import
import
import
import
import
import
import
import

android.app.Activity;
android.content.Intent;
android.database.Cursor;
android.graphics.Bitmap;
android.graphics.BitmapFactory;
android.graphics.Rect;
android.net.Uri;
android.os.Bundle;
android.provider.MediaStore;
android.util.Log;
android.view.Menu;
android.view.MenuItem;
android.view.View;
android.widget.Button;
android.widget.ImageView;

public class MainActivity extends Activity {


//
//
//
//
//

@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
}
@Override
public boolean onCreateOptionsMenu(Menu menu) {
// Inflate the menu; this adds items to the action bar if it is present.
getMenuInflater().inflate(R.menu.main, menu);
return true;
}
@Override
public boolean onOptionsItemSelected(MenuItem item) {
// Handle action bar item clicks here. The action bar will
// automatically handle clicks on the Home/Up button, so long
// as you specify a parent activity in AndroidManifest.xml.
int id = item.getItemId();
if (id == R.id.action_settings) {
return true;
}
return super.onOptionsItemSelected(item);

//=====================================================
=======================
protected static final String TAG = "OpenCVTest";
//
@Override
//
public void onResume()
//
{
//
super.onResume();
//
OpenCVLoader.initAsync(OpenCVLoader.OPENCV_VERSION_3_1_0, this,
mLoaderCallback);
//
}
private BaseLoaderCallback mLoaderCallback = new BaseLoaderCallback(this) {
@Override
public void onManagerConnected(int status) {
switch (status) {
case LoaderCallbackInterface.SUCCESS:
{
Log.i(TAG, "OpenCV loaded successfully");
} break;
default:
{
super.onManagerConnected(status);
} break;
}
}
};
private static int RESULT_LOAD_IMAGE = 1;
//public Mat img=new Mat();
//public Mat result=new Mat();
static {
if (!OpenCVLoader.initDebug()) {
// Handle initialization error
}
}
private BaseLoaderCallback mOpenCVCallBack = new BaseLoaderCallback(this)
{

@Override
public void onManagerConnected(int status) {
switch (status) {
case LoaderCallbackInterface.SUCCESS:
{
Log.i(TAG, "OpenCV loaded successfully");
// Create and set View
setContentView(R.layout.activity_main);
} break;
default:
{
super.onManagerConnected(status);
} break;
}
}
};

@Override
public void onResume()
{
super.onResume();
OpenCVLoader.initAsync(OpenCVLoader.OPENCV_VERSION_2_4_9, this,
mLoaderCallback);
}
/** Called when the activity is first created. */
@Override
public void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
//
if(!OpenCVLoader.initAsync(OpenCVLoader.OPENCV_VERSION_2_4_9,
this,mOpenCVCallBack))
//
Log.v(TAG, "Cannot connect to OpenCV Manager");
Button buttonLoadImage = (Button) findViewById(R.id.buttonLoadPicture);
buttonLoadImage.setOnClickListener(new View.OnClickListener() {
@Override
public void onClick(View arg0) {
Intent i = new Intent(
Intent.ACTION_PICK,
android.provider.MediaStore.Images.Media.EXTERNAL_CONTENT_URI);
}
}

startActivityForResult(i, RESULT_LOAD_IMAGE);

});

@Override
protected void onActivityResult(int requestCode, int resultCode, Intent data) {
super.onActivityResult(requestCode, resultCode, data);
if (requestCode == RESULT_LOAD_IMAGE && resultCode == RESULT_OK
&& null != data) {
Uri selectedImage = data.getData();
String[] filePathColumn = { MediaStore.Images.Media.DATA };
Cursor cursor = getContentResolver().query(selectedImage,
filePathColumn, null, null, null);
cursor.moveToFirst();
int columnIndex = cursor.getColumnIndex(filePathColumn[0]);
String picturePath = cursor.getString(columnIndex);
cursor.close();
ImageView imageView = (ImageView) findViewById(R.id.imgView);
Bitmap bmp=BitmapFactory.decodeFile(picturePath);
Log.i(TAG, picturePath);
Mat img=Highgui.imread(picturePath);
//Mat img=Imgcodecs.imread(picturePath);
Mat result=new Mat();
//Utils.bitmapToMat(bmp, img);
//Imgproc.cvtColor(img,result,Imgproc.COLOR_BGRA2BGR);

result=steptowatershed(img);
//Imgproc.cvtColor(result, img,Imgproc.COLOR_BGR2BGRA,4);
Utils.matToBitmap(result, bmp, true);
Log.i(TAG, "all okay");
imageView.setImageBitmap(bmp);
}

}
public Mat steptowatershed(Mat img)
{
Mat threeChannel = new Mat();
Imgproc.cvtColor(img, threeChannel, Imgproc.COLOR_BGR2GRAY);
Imgproc.threshold(threeChannel, threeChannel, 100, 255,
Imgproc.THRESH_BINARY);
Mat fg = new Mat(img.size(),CvType.CV_8U);
Imgproc.erode(threeChannel,fg,new Mat());
Mat bg = new Mat(img.size(),CvType.CV_8U);
Imgproc.dilate(threeChannel,bg,new Mat());
Imgproc.threshold(bg,bg,1, 128,Imgproc.THRESH_BINARY_INV);
Mat markers = new Mat(img.size(),CvType.CV_8U, new Scalar(0));
Core.add(fg, bg, markers);
Mat result1=new Mat();
WatershedSegmenter segmenter = new WatershedSegmenter();
segmenter.setMarkers(markers);
result1 = segmenter.process(img);
return result1;
}
public class WatershedSegmenter
{
public Mat markers=new Mat();
public void setMarkers(Mat markerImage)
{
}

markerImage.convertTo(markers, CvType.CV_32SC1);

public Mat process(Mat image)


{
Imgproc.watershed(image,markers);
markers.convertTo(markers,CvType.CV_8U);
return markers;
}
}

Main.xml
<RelativeLayout xmlns:android="http://schemas.android.com/apk/res/android"
xmlns:tools="http://schemas.android.com/tools"
android:layout_width="match_parent"
android:layout_height="match_parent"
android:paddingBottom="@dimen/activity_vertical_margin"
android:paddingLeft="@dimen/activity_horizontal_margin"
android:paddingRight="@dimen/activity_horizontal_margin"
android:paddingTop="@dimen/activity_vertical_margin"
tools:context="com.example.watershedsegmentation.MainActivity" >
<ImageView

android:id="@+id/imgView"
android:layout_width="fill_parent"
android:layout_weight="1"
android:layout_height="wrap_content"></ImageView>
<Button
android:id="@+id/buttonLoadPicture"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:layout_alignParentBottom="true"
android:layout_centerHorizontal="true"
android:layout_weight="0"
android:text="Ambil Citra" />
</RelativeLayout>

Vous aimerez peut-être aussi