Skip to content
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.

Sign up for GitHub

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

Jump to bottom

Allow 1 channel input? #17

Open
turian opened this issue Dec 18, 2020 · 2 comments
Open

Allow 1 channel input? #17

turian opened this issue Dec 18, 2020 · 2 comments

Comments

@turian
Copy link

turian commented Dec 18, 2020

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.
@turian
Copy link
Author

turian commented Apr 11, 2021

Would love to have this feature

@MingLin-home
Copy link
Collaborator

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@turian @MingLin-home