-
-
Notifications
You must be signed in to change notification settings - Fork 66
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
UNet inchannels != 3 not working #243
Comments
Hi @nickkeepfer, as I understand it, the way U-Net is being constructed right now requires you to specify UNet((128,128),1,3,Metalhead.backbone(DenseNet(121; inchannels = 1))) should hopefully yield the model that you want. Do let me know if this works the way you expect it to. If it does, there can be a discussion around reducing this redundancy. |
Yep, that's exactly what I needed, thanks! From a new user's perspective, it would be better if the inchannels from UNet was automatically propagated to the backbone, rather than having to specify it twice. Or at least making this clear in the docs with an example that you do need to specify it twice |
Yeah, it's definitely weird to keep it this way, I will try and land a PR soon to address this. |
Fix UNet implementation to support input with channel sizes other than 3
It seems the UNet implementation only works with inputs with channel-size 3
ERROR: DimensionMismatch: layer Conv((7, 7), 3 => 64, pad=3, stride=2, bias=false) expects size(input, 3) == 3, but got 128×128×1×1 Array{Flux.NilNumber.Nil, 4}
The text was updated successfully, but these errors were encountered: