UNet3plus_pth
UNet3 + / UNet ++ / UNet,用于Pytorth的深度自动人像抠像中
依存关系
Python 3.6
PyTorch> = 1.1.0
火炬视觉> = 0.3.0
未来0.18.2
matplotlib 3.1.3
的numpy 1.16.0
枕头6.2.0
protobuf 3.11.3
张量板1.14.0
tqdm == 4.42.1
数据
该模型从零开始使用18000张图像进行了训练(数据通过2000images进行了扩充)。训练数据集来自“ 。 您可以在百度云中下载 。 密码:ndg8仅用于学术交流,如果有报价,请告知原始作者!
我们通过旋转和缩放来增加图像的数量。 使用四个旋转角{-45°,−22°,22°,45°}和四个刻度{0.6、0.8、1.2、1.5}。 我们还应用了四个不同的Gamma变换来增加颜色变
1