madhurjindal's picture
Update requirements.txt
127ffca verified
torch
matplotlib
numpy
albumentations
opencv-python
pytorch_lightning
grad-cam