DORE HAIRSEGMENT

private var modelManager: HairSegmentManager?

modelManager = HairSegmentManager()
        
//load license
let isValid:Bool = (modelManager?.init_data(licKey: "insert your license key here") ?? false)

Step 3:

Execute Segment Session with input image

//run model and get result
let inputimage:UIImage = UIImage(pixelBuffer: pixelBuffer)!

//run model and get result
let outImage:CGImage = (self.modelManager?.run_model(onFrame: inputimage))!

//preview the output image
DispatchQueue.main.async {
   self.segmentView.image =  ciImage
}

Hair Color Change

//get color mask (you can adjust RGBA value)
let ciImage:UIImage = getColorMask(result.semanticPredictions,R: RGBCode[0], G: RGBCode[1], B: RGBCode[2], A: 255)!
        
let rgbCIimage:CIImage = CIImage.init(cvPixelBuffer: pixelBuffer)
let img:UIImage = convertCItoUIimage(cmage: rgbCIimage)
        
//you can change BlendMode and adjust alpha value (0.1 to 1)
let outputImage:UIImage = maskblendImage(backgroundImage: img, maskImage: ciImage, maskblendMode: CGBlendMode.multiply, blendAlpha: 0.8)!
        
        
DispatchQueue.main.async {
    self.maskoutView.image = outputImage
}

DoreHairSegmentLite (Lite Version SDK)

DoreHairSegmentLite is used for reducing app size, all support and library files will be downloaded at runtime (Internet connection required).

Total SDK size will be 1.2MB Max

How to use :

Step 1:

1. Add this line in your pod file

pod 'DoreHairSegmentLite'

2. Run pod install, and open project

pod install

3. Cocoapod version should be 1.12.1 or above

Step 2:

Add "HairSegmentLiteDelegate" in your view controller

class ImageSegment: UIViewController, HairSegmentLiteDelegate {

Initiate HairSegmentLiteManager

private var modelManager: HairSegmentLiteManager?

modelManager = HairSegmentLiteManager()
modelManager?.delegate = self

//Load license / start initiating
self.modelManager?.init_data(licKey: "insert your license key here")

Library file will be download at runtime when you initiate "HairSegmentLiteManager", it will be download and placed in app internal storage, you can get download completion signal through DoreHairSegmentLiteDelegate callback, use some progress indicator until download completion. this download progress will be processed only on first time, it will load with app internal storage location from second time load.

    func onHairSegmentLiteSuccess(_ info: String) {
        self.alertView.dismiss(animated: true, completion: nil)
        //DoreHairSegment Library files downloaded successfully...! Ready to run segment
        
    }
    
    func onHairSegmentLiteFailure(_ error: String) {
        self.alertView.dismiss(animated: true, completion: nil)
        //DoreHairSegment Library files downloading failed..!
        print(error)
    }
    
    func onHairSegmentLiteProgressUpdate(_ progress: String) {
        //DoreHairSegment Library files downloading...!
        print(progress)
        self.progressView!.progress = Float(progress)!
    }
    
    func onHairSegmentLiteDownloadSpeed(_ dps: String) {
        print(dps)
    }

Step 3:

Execute Segment Session with input image

//run and get result
let result:segmentOut  = segmentOut ( features: (self.modelManager?.run_model(onFrame: pixelBuffer))! )
        
//mask image Black - bacground, White - foreground
let ciImage:UIImage = getMaskBW(result.semanticPredictions)!
        
DispatchQueue.main.async {
   self.segmentView.image =  ciImage
}

Android

Technical Specifications:

Android Studio	3.5 or Above
Min SDK Version	API Level 23
Android Version	Marshmallow 6.0 or Above

You can implement DoreHairSegment SDK in your existing/new Android project with in just 3 Steps

Step 1:

1. Add this URL in project -> build.gradle (under respositories section)

allprojects {
    repositories {
        google()
        jcenter()
        maven { url "https://raw.github.com/doreai/Framework/master" }
    }
}

2. Add this line in app -> build.gradle (under dependencies section)

Version number will be vary depends on latest release

dependencies {
    ....
     
   implementation 'com.dore:hairsegment:1.0.7'

    
}

Don't forgot to Sync your project

Step 2:

Initiate DoreHairSegment Manager

Kotlin Code :

private val minThreshold = 0.4f  //change min Threshold From 0.1f to 0.6f for M1 model,  change min Threshold From 0.9 to 2.0f for M2 model
    private val maxThreshold = 1.01f //Only for M2 model - change max Threshold From 1.0f to 2.5f for M2 model


private val bEngine: HairSegmentManager = HairSegmentManager()

val lickeycode = "insert your license key here"
bEngine.init_data(this,lickeycode, minThreshold, "M1") //model_code : M1|M2

//====Model M1=====//
//Suggested for Picture segment
//================//

//====Model M2====//
//Suggested for Video Frame segment
//================//

Java Code :

HairSegmentManager bEngine = new HairSegmentManager();

String lickeycode = "insert your license key here";
bEngine.init_data(this,lickeycode, 0.4f, "M1");

Step 3:

Execute Segment Session with input image ( Get segment mask image )

Kotlin Code :

//get input bitmap image
val dimage = DoreImage.fromBitmap(tex_view.bitmap)
//get result
var result = bEngine.run(dimage)

runOnUiThread {
  //get segment mask image
  
  //M2 model - get black&white mask image (white - hair, black - other)
  //outImg.setImageBitmap(result?.getHairSegmentMask(1f))   

  //M1 Model - get black&white mask image (white - other, black - hair)
  outImg.setImageBitmap(result?.getMask(1f))  //For m1 model


}

Java Code :

//get input bitmap image
DoreImage dimage = DoreImage.fromBitmap(tex_view.bitmap);
//get result
DoreVisionSegmentResult result = bEngine.run(dimage);

runOnUiThread(new Runnable() {
   @Override
   public void run() {
    //outImg.setImageBitmap(result.getHairSegmentMask(1f)); //M2 model  
    outImg.setImageBitmap(result.getMask(1f));  //M1 model
  }
});

Change Hair Color

Kotlin Code :

val mask_0 = result!!.getHairSegmentMask(0.4f, cur_color, 1.0f)
val result_out = bEngine.clor_blend(org_img.bitmap,mask_0,PorterDuff.Mode.ADD)
        
runOnUiThread {
     bg_outImg.setImageBitmap(result_out)
}

Java Code :

Bitmap mask_0 = result.getMask(0.4f, cur_color, 1.0f);
Bitmap result_out = bEngine.clor_blend(org_img,bitmap,mask_0, PorterDuff.Mode.ADD);

runOnUiThread(new Runnable() {
  @Override
  public void run() {
    outImg.setImageBitmap(result_out);
  }
});

DoreHairSegmentLite (Lite Version SDK)

DoreHairSegmentLite is used for reducing app size, all support and library files will be downloaded at runtime (Internet connection required).

Total SDK size will be less than 1MB

How to use :

1. Add this URL in project -> build.gradle (under respositories section)

allprojects {
    repositories {
        google()
        jcenter()
        maven { url "https://raw.github.com/doreai/Framework/master" }
    }
}

2. Add this line in app -> build.gradle (under dependencies section)

Version number will be vary depends on latest release

dependencies {
    ....
     
   implementation 'com.dore:hairsegmentlite:1.0.4'

    
}

Don't forgot to Sync your project

Step 2:

Initiate DoreHairSegmentLite Manager

Kotlin Code :

private val bEngine: HairSegmentLiteManager = HairSegmentLiteManager()

val lickeycode = "insert your license key here"
bEngine.init_data(this,lickeycode)

Java Code :

HairSegmentLiteManager bEngine = new HairSegmentLiteManager();

String lickeycode = "insert your license key here";
bEngine.init_data(this,lickeycode);

Library file will be download at runtime when you initiate "HairSegmentLiteManager", it will be download and placed in app internal storage, you can get download completion signal through HairSegmentLiteListener callback, use some progress indicator until download completion. this download progress will be processed only on first time, it will load with app internal storage location from second time load.

Kotlin Code :

 bEngine.setHairSegmentLiteListener(object : HairSegmentLiteListener {

 override fun onSuccess(info: String) {
     //library file download completed...!
     isLibLoaded  = true
 }

 override fun onFailure(error: String) {

 }

 override fun onProgressUpdate(progress: String) {
    Log.d("Download Library : ", progress)
 }


 })

Java Code:

bEngine.setHairSegmentLiteListener(new HairSegmentLiteListener() {
   @Override
   public void onSuccess(String s) {
   //library files downloaded successfully...! ready to run
     isLibLoaded = true;
   }

   @Override
   public void onFailure(String s) {

   }

   @Override
   public void onProgressUpdate(String s) {
      Log.d("Download Library : ", s);
   }
 
});

Step 3:

Execute Segment Session with input image

Kotlin Code :

 if(isLibLoaded) {
      val dimage = DoreImage.fromBitmap(tex_view.bitmap)
      var result = bEngine.run(dimage)

      runOnUiThread {
          outImg.setImageBitmap(result?.getMask(1f))  //alpha value 0.1 to 1
       }
        
}

Java Code :

if(isLibLoaded) {
   //get input bitmap image
   DoreImage dimage = DoreImage.fromBitmap(cur_pic);
   //get result
   DoreVisionSegmentResult result = bEngine.run(dimage);

   runOnUiThread(new Runnable() {
     @Override
     public void run() {
      outImg.setImageBitmap(result.getMask(1f));     //alpha 0.1 to 1
     }
   });
}

Additional Note: Supported ABIs : armeabi-v7a and arm64-v8a, which should cover most modern Android devices.

Assistance

DoreHairSegment

IOS

DoreHairSegmentLite (Lite Version SDK)

Android

DoreHairSegmentLite (Lite Version SDK)

Company

License

Product

Help