requirements.txt missing library

[yassineAlouini]

It seems that the requirements.txt file is missing some of the necessary libraries. Besides, some aren't necessary (such as os and warnings, which come in the base library).

I can contribute an MR that adds these. 👌

[yassineAlouini]

The following requirements seem to work:

matplotlib
tqdm
numpy
opencv-python
torch
torchvision

[Skwarson96]

I already did PR #18

[AustinStarnes]

To be clear, os and warnings in the requirements file isn't just "unnecessary", but actually break the installation instruction pip install -r requirements.txt which was provided in the readme, since these modules are not provided by pip.