prediction about 2d-denseUnet about h-denseunet HOT 5 OPEN

The liver area is already generated and you can download "livermask". The 2d-denseUnet model used to produce accurate liver and tumor mask.

@hongson23 yes, you can.

from h-denseunet.

Hello @xmengli999, I have the same question. Could we predict liver segmentation using 2d-denseUnet?
Thank you

from h-denseunet.

thanks for your reply. because we want to have a Transfer learning based on your dataset. so we want to know how can we use 2d-model to produce liver mask or predict the liver areas. thanks a lot

from h-denseunet.

To produce the liver segmentation, I suggest you to use a standard 2D ResNet/denseunet model and you can get a satisfactory results(at least higher than 90%). You can remove the tumor label, and change the channel of the last convolution layer to 2.

from h-denseunet.

ok~ i get your suggestion , change the class from 3 to 2 . but how to write the prediction function? load the trained model and load the "predict_window_mulgpu" in the ./lib/func.py ?

from h-denseunet.