bioimage-io / spec-bioimage-io

Specification for the bioimage.io model description file.
https://bioimage-io.github.io/spec-bioimage-io/
MIT License
18 stars 17 forks source link

Request to allow `outputs` in `ModelDescr` to be `Sequence[Sequence[OutputTensorDescr]]` #615

Open qin-yu opened 5 months ago

qin-yu commented 5 months ago

Currently:

    outputs: NotEmpty[Sequence[OutputTensorDescr]]
    """Describes the output tensors."""

Cellpose's ResNet produces tuple[Tensor, Tensor, list[Tensor]]. It would be nice to support this if it only introduces little trouble, otherwise please ignore this because the list can be unpacked by who uploads the Cellpose models.