danielaivanova/

damaged-media

ARTeFACT: Benchmarking Segmentation Models on Diverse Analogue Media Damage

Here we provide example code for downloading the data, loading it as a PyTorch dataset, splitting by material and/or content, and visualising examples.

!pip install datasets !pip install -qqqU wandb transformers pytorch-lightning==1.9.2 albumentations torchmetrics torchinfo !pip install -qqq requests gradio

import os from glob import glob

import cv2… See the full description on the dataset page: https://huggingface.co/datasets/danielaivanova/damaged-media.