Allow 1 channel input?

Question

Allow 1 channel input?

turian opened this issue 4 years ago · comments

Can you please support choosing the number of channels? I have 1 channel input.

EfficientNet allows you to specify the number of input channels, for example if the image is grayscale.

https://github.com/lukemelas/EfficientNet-PyTorch/blob/master/efficientnet_pytorch/model.py#L321
in_channels is 3 by default but can be set to one.

Joseph Turian · Answer 1 · Sun Apr 11 2021 21:31:16 GMT+0800 (China Standard Time)

Would love to have this feature

Ming Lin · Answer 2 · Mon Sep 06 2021 11:32:03 GMT+0800 (China Standard Time)

Sorry for the late reply!

This project is obsoleted and the development is merged into our ICCV 2021 work Zen-NAS. The new Github repository is https://github.com/idstcv/ZenNAS.