numpy
opencv-python-headless
scikit-image
networkx
shapely
Pillow
torch
torchvision