
UNet3plus_pth是UNet3 + UNet ++ UNet,应用于PyTorch深度自动人像抠像。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
该模型,命名为UNet3plus_pth,以及其变体UNet3+、UNet++和UNet,主要应用于PyTorch框架中的深度自动人像抠像任务。其运行依赖于Python 3.6环境,并需要PyTorch版本大于等于1.1.0,火炬视觉库版本大于等于0.3.0,未来库版本大于等于0.18.2,matplotlib 3.1.3,numpy 1.16.0,枕头(PIL)库6.2.0,protobuf 3.11.3以及TensorBoard 1.14.0和tqdm==4.42.1。模型训练所使用的数据集包含18000张图像,并通过扩充方法使用2000张图像进一步增加数据量。扩充策略包括旋转和缩放操作,具体而言使用了四个旋转角度{-45°,−22°,22°,45°}以及四个不同的缩放比例{0.6、0.8、1.2、1.5}。此外,还应用了四个不同的Gamma变换以丰富图像的色彩多样性。数据集的来源位于“ ” (请自行搜索),您可以在百度云上找到该数据集下载链接。密码为:ndg8。请注意此模型仅供学术交流使用;若有引用时务必告知原始作者并进行相应的报价!
全部评论 (0)
还没有任何评论哟~


