这只是个小 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)
这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。
https://study.congcong.us/t/363546
V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。
V2EX is a community of developers, designers and creative people.