numpy matplotlib tqdm gradio torch torchmetrics torchvision segmentation_models_pytorch scikit-image imageio medpy einops loguru IPython