import albumentations
import numpy as np
import torch
transform = albumentations.Compose(
     [
         albumentations.Resize(480, 480),
         albumentations.HorizontalFlip(p=1.0),
         albumentations.RandomBrightnessContrast(p=1.0),
     ],
     bbox_params=albumentations.BboxParams(format="coco", label_fields=["category"]),
 )