文件名称:Pytorch下实现Unet对自己多类别数据集的语义分割
文件大小:69KB
文件格式:ZIP
更新时间:2024-02-01 03:03:27
pytorch Unet 语义分割
Unet通常应用到单类别的语义分割,经过调整后该代码适合于多类别的语义分割。对应博客:https://blog.csdn.net/brf_UCAS/article/details/112383722
【文件预览】:
U-Net
----dataloaders()
--------utils.py(3KB)
--------__pycache__()
--------custom_transforms.py(5KB)
--------__init__.py(2KB)
--------datasets()
----data()
--------test.txt(800B)
--------info.json(976B)
----train.py(12KB)
----demo.py(4KB)
----mypath.py(497B)
----modeling()
--------__pycache__()
--------__init__.py(0B)
--------unet.py(3KB)
----curve()
--------loss.txt(0B)
----utils()
--------loss.py(2KB)
--------summaries.py(1KB)
--------lr_scheduler.py(3KB)
--------metrics.py(2KB)
--------__pycache__()
--------calculate_weights.py(985B)
--------saver.py(2KB)