fepegar / highresnet

PyTorch implementation of HighRes3DNet
MIT License
49 stars 15 forks source link

add support for selective dilation #39

Open dvolgyes opened 2 years ago

dvolgyes commented 2 years ago

Description

It would be very nice to replace the current dilation options with a more intelligent one. Currently the number of dilations can be specified, and the dilations always increase. There are multiple problems with this, but the two largest ones:

So instead of the current integer input it would be nice to accept two kind of inputs:

There are good arguments that a final refinement could be very helpful to remove small scale noise. Also, the receptive field is better tunable this way. The implementation is relatively straightforward.

fepegar commented 2 years ago

Hi, @dvolgyes. Sounds good. Would you like to contribute with a PR?