the dataset needs to be in the same directory as the code.
The following steps need to be implemented to create an environment for Task 1:
a) python3 -m venv env
b) source env/bin/activate
c) pip install -r requirementsTASK1.txt
to run the code, python3 <filename>.py
The code was run using the weights from the models for ease of visualisation and in order to run the code, extract the weights of the trained models and replace them in the 'MODEL_CONFIGS'