Hello, I believe that you have a mistake in the res_block implementation:
https://github.com/kim-younghan/MultiResUNet/blob/1c4a77973c18ada6dab009ed603682dfdd20c54d/multiresunet/model/MultiResUNet.py#L72-L73
You forward the input into the same block a couple of times, however, we need to start with, 'stage' num blocks, then run the input through each one of them.
Best Regards,
Eyal
Hello, I believe that you have a mistake in the res_block implementation:
https://github.com/kim-younghan/MultiResUNet/blob/1c4a77973c18ada6dab009ed603682dfdd20c54d/multiresunet/model/MultiResUNet.py#L72-L73
You forward the input into the same block a couple of times, however, we need to start with, 'stage' num blocks, then run the input through each one of them.
Best Regards,
Eyal