V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
爱意满满的作品展示区。
lingoerer
V2EX  ›  分享创造

在 iOS 上用深度网络把黑白视频变彩色~~

  •  
  •   lingoerer · 2017-05-24 19:20:48 +08:00 · 3366 次点击
    这是一个创建于 2739 天前的主题,其中的信息可能已经有所发展或是发生改变。
    这只是个小 Demo,代码地址在这里(恬不知耻地骗 Star 来了……):

    https://github.com/lingoer/MetalColor

    使用方法,build 起来以后,拿摄像头对着黑白的图片或者电影( [地道战] 效果不咋好,但 [平原游击队] 还行),手机屏幕上显示的是自动着色后的画面。从代码来说,基本上,也就是拿 TensorFlow 训练模型,然后把模型在 iOS 上面用 Metal 重新实现一遍,并把 inference 跑起来。

    事实上,这东西其实是我为了应付上上周在第二届 Swift 大会上面的演讲做的,当时的演讲主要为了让大家把 Metal 和深度学习联系起来加以重视,因为深度学习的 Inference 阶段今后会越来越多地跑在移动设备上面(正好 Google 就跳出个 Google Lens 出来佐证)。结果演讲当天悲剧地并没有跑起来代码,也就没法现场演示,小遗憾……所以这里把代码开出来,让大家可以在自己的电脑上 build 出来玩儿。

    差点儿忘了,当时演讲的 Slides 在这里:
    https://github.com/atConf/atswift-2017-resources/tree/master/METAL%20%26%20DEEP%20LEARNING

    关于 iOS 上的深度学习,这次大会上还有尹航大神也在讲,是更多关于 TensorFlow 的,非常非常精彩,强烈推荐大家也把那个 Slides 下下来看看。

    效果如下(渣美工,就只扔了俩 View,只是演示代码,轻喷):


    !( https://github.com/lingoer/MetalColor/raw/master/result1.jpeg)


    !( https://github.com/lingoer/MetalColor/raw/master/result2.jpeg)
    7 条回复    2017-05-30 13:52:50 +08:00
    ZernonHuang
        1
    ZernonHuang  
       2017-05-24 19:56:27 +08:00
    我记得搞 iOS 搞深度学习是有 Xcode 版本限制,和机型限制。你这个有吗
    ZernonHuang
        2
    ZernonHuang  
       2017-05-24 20:01:41 +08:00
    呃,刚去看了,readme 上写了限制,不好意思没仔细看就问了
    StargazerWikiv
        3
    StargazerWikiv  
       2017-05-25 11:54:10 +08:00
    iPhone 7 Plus 不能 run ?获取不了视频流的 view.currentDrawable 直接报错为 nil... 看过设置已经开了照相机权限了。
    lingoerer
        4
    lingoerer  
    OP
       2017-05-25 12:03:51 +08:00
    @StargazerWikiv 估计需要杀掉重新跑吧~~~理论上 7P 应该是没有问题的
    lingoerer
        5
    lingoerer  
    OP
       2017-05-25 12:05:45 +08:00
    @StargazerWikiv 是不是 iOS 的版本木有对?
    StargazerWikiv
        6
    StargazerWikiv  
       2017-05-25 13:36:58 +08:00
    @lingoerer 10.2 Demo 是 10.3 我改了下
    googlecom
        7
    googlecom  
       2017-05-30 13:52:50 +08:00
    深度网络高大上
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   2837 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 23ms · UTC 14:04 · PVG 22:04 · LAX 06:04 · JFK 09:04
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.