org.deeplearning4j.zoo.model

Class UNet

java.lang.Object

org.deeplearning4j.zoo.ZooModel

org.deeplearning4j.zoo.model.UNet

All Implemented Interfaces:

InstantiableModel

public class UNet
extends ZooModel

U-Net An implementation of U-Net, a deep learning network for image segmentation in Deeplearning4j. The u-net is convolutional network architecture for fast and precise segmentation of images. Up to now it has outperformed the prior best method (a sliding-window convolutional network) on the ISBI challenge for segmentation of neuronal structures in electron microscopic stacks.

Paper: https://arxiv.org/abs/1505.04597

Weights are available for image segmentation trained on a synthetic dataset

Author:

Justin Long (crockpotveggies)

Method Summary

Modifier and Type	Method and Description
ComputationGraphConfiguration.GraphBuilder	graphBuilder()
ComputationGraph	init()
ModelMetaData	metaData()
Class<? extends Model>	modelType()
long	pretrainedChecksum(PretrainedType pretrainedType)
String	pretrainedUrl(PretrainedType pretrainedType)
void	setInputShape(int[][] inputShape)

Methods inherited from class org.deeplearning4j.zoo.ZooModel

initPretrained, initPretrained, modelName, pretrainedAvailable

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

pretrainedUrl

public String pretrainedUrl(PretrainedType pretrainedType)

pretrainedChecksum

public long pretrainedChecksum(PretrainedType pretrainedType)

modelType

public Class<? extends Model> modelType()

init

public ComputationGraph init()

graphBuilder

public ComputationGraphConfiguration.GraphBuilder graphBuilder()

metaData

public ModelMetaData metaData()

setInputShape

public void setInputShape(int[][] inputShape)