torch
scikit-image
matplotlib