Efficient backbones is the project which aims to provide efficient SOTA vision backbones based on PyTorch.
Currently following models are supported;
NOTE: All models implemented in this project doesn't have head.
pip install git+https://github.com/ozora-ogino/efficient_backbones
All model have out_channels
attribute in order to access the number of out channels of backbone in the same way.
from efficient_backbones import build_backbone
resnet_backbone = build_backbone("resnet18")
out_channels = resnet_backbone.out_channels