The repository contains 3D variants of EfficientNet models for classification.
This repository is based on great efficientnet repo by @qubvel
- keras >= 2.2.0 and tensorflow >= 1.13
- keras_applications >= 1.0.7
pip install efficientnet-3D
# models can be build with Keras or Tensorflow frameworks
# use keras and tfkeras modules respectively
# efficientnet.keras / efficientnet.tfkeras
import efficientnet_3D.keras as efn
# import efficientnet_3D.tfkeras as efn
model = efn.EfficientNetB0(input_shape=(64, 64, 64, 3), weights='imagenet')
- https://github.com/qubvel/classification_models - original classification 2D repo
- https://github.com/qubvel/segmentation_models - original segmentation 2D repo
- classification_models_3D - models for classification in 3D
- volumentations - 3D augmentations
- There is no DepthwiseConv3D layer in keras, so repo used custom layer from this repo by @alexandrosstergiou.