文件名称：pytorch_segmentation：在PyTorch中实现的语义分割模型，数据集和损失
文件大小：598KB
文件格式：ZIP
更新时间：2024-02-25 18:06:01
computer-vision deep-learning semantic-segmentation DeeplearningPython
PyTorch中的语义分割 此仓库包含一个PyTorch，用于不同数据集的不同语义分割模型的实现。 要求 在运行脚本之前，需要先安装PyTorch和Torchvision，以及用于数据预处理的PIL和opencv和用于显示培训进度的tqdm 。 支持PyTorch v1.1（使用新的受支持的Tensoboard）； 可以使用更早期的版本，但不要使用tensoboard，而要使用tensoboardX。 pip install -r requirements.txt 或本地安装 pip install --user -r requirements.txt 主要特点 清晰易用的结构， 一个j
【文件预览】：
pytorch_segmentation-master
----inference.py(7KB)
----train.py(2KB)
----utils()
--------helpers.py(2KB)
--------sync_batchnorm()
--------torchsummary.py(3KB)
--------__init__.py(26B)
--------lovasz_losses.py(8KB)
--------metrics.py(2KB)
--------losses.py(3KB)
--------lr_scheduler.py(4KB)
--------palette.py(8KB)
--------transforms.py(10KB)
--------logger.py(319B)
----models()
--------segnet.py(9KB)
--------gcn.py(11KB)
--------pspnet.py(8KB)
--------__init__.py(318B)
--------unet.py(8KB)
--------deeplabv3_plus.py(13KB)
--------fcn.py(5KB)
--------duc_hdc.py(9KB)
--------resnet.py(11KB)
--------enet.py(9KB)
--------upernet.py(6KB)
----images()
--------tb2.png(345KB)
--------colour_scheme.png(30KB)
--------learning_rates.png(54KB)
--------tb1.png(95KB)
----base()
--------base_trainer.py(8KB)
--------__init__.py(115B)
--------base_dataloader.py(3KB)
--------base_dataset.py(6KB)
--------base_model.py(893B)
----requirements.txt(112B)
----dataloaders()
--------ade20k.py(2KB)
--------labels()
--------cityscapes.py(4KB)
--------coco.py(4KB)
--------__init__.py(105B)
--------voc.py(4KB)
----LICENSE(1KB)
----config.json(2KB)
----README.md(14KB)
----.gitignore(1KB)
----trainer.py(8KB)