khdlr/HED-UNet

Testing data

Malaika68 opened this issue · 7 comments

After training.. how can we test our data?

Can you please help in tiling and stitching data?

khdlr commented

Hi Malaika, this paper might help: https://arxiv.org/abs/1805.12219

Hey thanks for the help. dimensions of my dataset image are 57x60. but it does not train on that dimension. Can ypu please tell where i can change the dimensions in the code?

Also please tell me that why it is cropping my image from side after training?

khdlr commented

That's because the input dimensions should be a multiple of 32 (this is a limitation that all UNet-based models share in a similar fashion).

You could try padding the inputs to a size of 64x64.

Okay great.. I have tried that. Also can you please tell the channels of images and its labels. Because I am getting channels dimension error. If the channels of images and labels are 3 or 1?

khdlr commented

You can set the number of input channels in config.yml (model_args.input_channels), it should be 3 by default. Labels should have 1 as the channel dimension.