Spaces:

XDL-Collitis
/

XDL-Colitis-Demo

Runtime error

XDL-Colitis-Demo / src /preprocessing.py

Simplify DenseNet, preprocessing, and DenseNet-only XDL helpers

710f946 9 months ago

523 Bytes

from typing import Sequence

import torchvision.transforms as transforms


def preprocess(target_input_size: Sequence[int]) -> transforms.Compose:
    """Return the inference transform used by the demo model."""
    if not (isinstance(target_input_size, (list, tuple)) and len(target_input_size) == 3):
        raise ValueError("target_input_size must be (C, H, W)")

    _, height, width = target_input_size
    return transforms.Compose([
        transforms.Resize((height, width)),
        transforms.ToTensor(),
    ])

	from typing import Sequence

	import torchvision.transforms as transforms


	def preprocess(target_input_size: Sequence[int]) -> transforms.Compose:
	"""Return the inference transform used by the demo model."""
	if not (isinstance(target_input_size, (list, tuple)) and len(target_input_size) == 3):
	raise ValueError("target_input_size must be (C, H, W)")

	_, height, width = target_input_size
	return transforms.Compose([
	transforms.Resize((height, width)),
	transforms.ToTensor(),
	])