ECAENet (TensorFlow and Keras)

ECAENet: EfficientNet with Efficient Channel Attention for Plant Species Recognition

(SCI:Q3) (Journal of Intelligent & Fuzzy Systems)

Results

ECAENet Architecture

Requirements

pip install pillow

pip install matplotlib

pip install scikit-learn

pip install tensorflow-addons

pip install tensorflow-gpu

conda install cudnn

Start Up

You can run the train.py to train or evaluate as follow:

python train.py 
python train.py --epoch 100 --lr 5e-5 --data flower --af swish --at eca --batch_size 32 --img_size 224

Contact Information

If you have any suggestion or question, you can leave a message here or contact us directly: [email protected]. Thanks for your attention!