登录 注册
当前位置:主页 > 资源下载 > 12 > 快速神经风格化的pytorch实现,采用MobileNetV2的瓶颈模块

快速神经风格化的pytorch实现,采用MobileNetV2的瓶颈模块

  • 更新:2024-12-11 23:23:08
  • 大小:4.17MB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

带有MobileNetV2瓶颈块的快速神经风格 该存储库包含用于艺术风格转换的算法的PyTorch实现。 该实现基于以下文件和存储库: 与其他实现的主要区别 残差块和卷积更改为MobileNetV2瓶颈块,该瓶颈块使用了“反向残差”和“深度可分离卷积”。 在图片上,您可以看到两种类型的MobileNetV2瓶颈块。 使用左一个代替残差块,使用右一个代替卷积层。 更改目的: 将变压器网络的可训练参数数量从〜1.67m减少到〜0.23m ,因此减少了变压器网络使用的内存量。 从理论上讲,这应该在训练期间,并且更重要的是在推理期间(快速神经风格应尽可能快)提供良好的加速。 看来实际上情况并不是很好,并且变压器网络的这种体系结构仅比原始变压器网络快一点。 造成这种情况的主要原因是,深度卷积在GPU上的实现不如普通卷积有效(在CPU上,速度提高了,但仍然不是很大)。 此实现使用围绕PyTor