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