unetdl4j

Training Unet with DL4J

Basic use of DL4J for training a Unet network.

Use of zhixuhao Github ((https://github.com/zhixuhao/unet)) example for data/images.

How to compile

% mvn compile
% mvn package   # create jar

How to run

% java -cp target/segmentation-1.0-SNAPSHOT-bin.jar org.sbml.spatial.segmentation.TrainUnetModel