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变换来增加颜色变
2021-07-27 20:28:54 3.58MB pytorch unet portrait-matting unet-plusplus
1