You need to have folowing requirments to run this code smoothly,
- Python 3.5+
- Python numpy package
- Tensorflow 2.0
- Keras 2.3.1
- Efficientnet : https://github.com/qubvel/efficientnet
- Classification models : https://github.com/qubvel/classification_models.git
- Segmentation models : https://github.com/qubvel/segmentation_models
- Albumentation : https://github.com/albu/albumentations
- General python packages, namely: numpy, argparse, matplotlib, os, sys, cv2, skimage, sklearn, random, re, PIL, pathlib, shutil assumed to be installed