This repository contains 5 directories:

• data (pickle file used to train the model)
• code (python script for training)
• notebook (jupyter notebook to visualize data and results)
• results (containing all predicted results)
• weights (trained weights for CNN models)

You can find all the data, results and weights here

Python 3.8, Tensorflow 2.2