Spaces:
Running
Running
File size: 566 Bytes
aea73e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
# -*- coding: utf-8 -*-
# Graph Data model
#
# For more information, please check out docs/readmes/graphs.md
#
# @ Fabian Hörst, [email protected]
# Institute for Artifical Intelligence in Medicine,
# University Medicine Essen
from dataclasses import dataclass
from typing import List
import torch
from datamodel.graph_datamodel import GraphDataWSI
@dataclass
class CellGraphDataWSI(GraphDataWSI):
"""Dataclass for Graph Data
Args:
contours (List[torch.Tensor]): Contour Data for each object.
"""
contours: List[torch.Tensor]
|