颜色分类leetcode unet nested multiple classification 该存储库包含基于UNet和UN...

颜色分类leetcode Unet和Unet++：使用Pytorch进行多重分类该存储库包含基于和的多分类图像分割模型的代码用法注意：使用Python 3数据集确保将文件按以下结构放置： data ├── images | ├── 0a7e06.jpg │ ├── 0aab0a.jpg │ ├── 0b1761.jpg │ ├── ... | └── masks ├── 0a7e06.png ├── 0aab0a.png ├── 0b1761.png ├── ... mask是单通道类别索引。例如，你的数据集有三个类别，掩码应该是8位图像，值为0,1,2作为分类值，这个图像看起来是黑色的。演示数据集您可以从to data/下载演示数据集训练python train.py推理python inference.py -m ./data/checkpoints/epoch_10.pth -i ./data/test/input -o ./data/test/output如果你想用颜色突出你的面具，你可以python inference_co