pytorch_segmentation:在PyTorch中实现的语义分割模型数据集和损失 源码
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.zip
(预估有个57文件)
pytorch_segmentation-master
inference.py
7KB
train.py
2KB
utils
helpers.py
2KB
sync_batchnorm
batchnorm_reimpl.py
2KB
comm.py
4KB
__init__.py
507B
replicate.py
3KB
暂无评论