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.