Pytorch-UNet:U-Net的PyTorch实施,用于高质量图像的图像语义分割-源码

上传者: u013546588 | 上传时间: 2022-04-06 20:59:31 | 文件大小: 26KB | 文件类型: ZIP
UNet:使用PyTorch进行语义分割 在PyTorch中针对高清晰度图像针对Kaggle的自定义实施 。 该模型是从头开始训练的,具有5000张图像(无数据增强),并且在超过100k张测试图像上获得了0.988423(735中的511)的。 可以通过更多的培训,数据增强,微调,使用CRF后处理以及在蒙版边缘上施加更多权重来提高此分数。 Carvana数据可在上。 用法 注意:使用Python 3.6或更高版本 预言 训练好模型并将其保存到MODEL.pth后,您可以通过CLI轻松测试图像上的输出蒙版。 预测单个图像并保存: python predict.py -i image.jpg -o output.jpg 要预测多幅图像并显示它们而不保存它们: python predict.py -i image1.jpg image2.jpg --viz --no-save >

文件下载

资源详情

[{"title":"( 17 个子文件 26KB ) Pytorch-UNet:U-Net的PyTorch实施,用于高质量图像的图像语义分割-源码","children":[{"title":"Pytorch-UNet","children":[{"title":".gitignore <span style='color:#111;'> 67B </span>","children":null,"spread":false},{"title":"unet","children":[{"title":"__init__.py <span style='color:#111;'> 29B </span>","children":null,"spread":false},{"title":"unet_parts.py <span style='color:#111;'> 2.52KB </span>","children":null,"spread":false},{"title":"unet_model.py <span style='color:#111;'> 1.13KB </span>","children":null,"spread":false}],"spread":true},{"title":"requirements.txt <span style='color:#111;'> 66B </span>","children":null,"spread":false},{"title":"data","children":[{"title":"masks","children":[{"title":".keep <span style='color:#111;'> 0B </span>","children":null,"spread":false}],"spread":true},{"title":"imgs","children":[{"title":".keep <span style='color:#111;'> 0B </span>","children":null,"spread":false}],"spread":true}],"spread":true},{"title":"eval.py <span style='color:#111;'> 1.05KB </span>","children":null,"spread":false},{"title":"LICENSE <span style='color:#111;'> 34.32KB </span>","children":null,"spread":false},{"title":"utils","children":[{"title":"data_vis.py <span style='color:#111;'> 508B </span>","children":null,"spread":false},{"title":"dataset.py <span style='color:#111;'> 2.27KB </span>","children":null,"spread":false}],"spread":true},{"title":"README.md <span style='color:#111;'> 5.35KB </span>","children":null,"spread":false},{"title":"submit.py <span style='color:#111;'> 1.38KB </span>","children":null,"spread":false},{"title":"hubconf.py <span style='color:#111;'> 560B </span>","children":null,"spread":false},{"title":"predict.py <span style='color:#111;'> 4.14KB </span>","children":null,"spread":false},{"title":"dice_loss.py <span style='color:#111;'> 1.20KB </span>","children":null,"spread":false},{"title":"train.py <span style='color:#111;'> 7.36KB </span>","children":null,"spread":false}],"spread":false}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明