efficientnet-3D

EfficientNet models in 3D variant for keras and TF.keras


License
MIT
Install
pip install efficientnet-3D==1.0.1

Documentation

EfficientNet 3D Keras (and TF.Keras)

The repository contains 3D variants of EfficientNet models for classification.

This repository is based on great efficientnet repo by @qubvel

Requirements

  • tensorflow >= 2.3.2
  • keras_applications >= 1.0.8

Installation

pip install efficientnet-3D

Examples

Loading model:
# 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')

Related repositories

Unresolved problems