Spaces:

uva-cv-lab
/

SAB3R

Runtime error

SAB3R / eval /utils.py

Initial SAB3R demo release

c7b663e verified 6 months ago

346 Bytes

import torch
import numpy as np
import torchvision

default_seg_target_transform = torchvision.transforms.Compose(
    [
        torchvision.transforms.Resize((224, 224)),
        torchvision.transforms.Lambda(lambda x: torch.tensor(np.array(x))),
    ]
)


class DummyArgs:
    def __init__(self, **kwargs):
        self.__dict__.update(kwargs)

	import torch
	import numpy as np
	import torchvision

	default_seg_target_transform = torchvision.transforms.Compose(
	[
	torchvision.transforms.Resize((224, 224)),
	torchvision.transforms.Lambda(lambda x: torch.tensor(np.array(x))),
	]
	)


	class DummyArgs:
	def __init__(self, **kwargs):
	self.__dict__.update(kwargs)