Dependencies torch torchvision numpy opencv matplotlib Installation You can install the required packages using the following: pip install torch torchvision numpy opencv-python matplotlib