sergivalverde / nicMSlesions

Easy multiple sclerosis white matter lesion segmentation using convolutional deep neural networks.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Inference segmentation from de baseline_2ch

isabellameds opened this issue · comments

Dear Sergi,

I have a question regarding the inference process, is it possible to predict the lesions using just your pretrained model? Or how to predict using only a full trained model?
I tried to do that today, but the results are weird, I think I may be writing something wrong in the configuration file.
I used the following parameters regarding the model:

[train]
full_train = True
pretrained_model = None

[model]
name = baseline_2ch
pretrained = None
train_split = 0.25
max_epochs = 400
patience = 50
batch_size = 128
net_verbose = 1
gpu_number = 0

Is there something wrong?

Kind regards,
Isabella