jcreinhold / tiramisu-brulee

A 2D and 3D PyTorch implementation of the Tiramisu CNN
Other
5 stars 2 forks source link

Support multi-class segmentation #4

Closed jcreinhold closed 3 years ago

jcreinhold commented 3 years ago

The current version of lesionseg assumes a binary segmentation task. Extend the framework to support multi-class segmentation.

jcreinhold commented 3 years ago

Added with 6d96cf12c4f9bed0b0454e93f0a69d060e3d5a9d