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

[lingoerer]

这只是个小 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)

[ZernonHuang]

我记得搞 iOS 搞深度学习是有 Xcode 版本限制，和机型限制。你这个有吗

[ZernonHuang]

呃，刚去看了，readme 上写了限制，不好意思没仔细看就问了

[StargazerWikiv]

iPhone 7 Plus 不能 run ？获取不了视频流的 view.currentDrawable 直接报错为 nil... 看过设置已经开了照相机权限了。

[lingoerer]

@StargazerWikiv 估计需要杀掉重新跑吧~~~理论上 7P 应该是没有问题的

[lingoerer]

@StargazerWikiv 是不是 iOS 的版本木有对？

[StargazerWikiv]

@lingoerer 10.2 Demo 是 10.3 我改了下

[googlecom]

深度网络高大上