深度学习（六十七）metal forge深度学习库使用

1、设置输入：

let input = Input()

或者

let input = Input(width: 100, height: 100, channels: 3)

2、创建网络：

let output = input--> Resize(width: 28, height: 28)--> Convolution(kernel: (5, 5), channels: 20, activation: relu, name: "conv1")--> Dense(neurons: 10, name: "dense1")--> Softmax()

3、链接网络、加载参数

model = Model(input: input, output: output)

let success = model.compile(device: device, inflightBuffers: 3) {name, count, type in return ParameterLoaderBundle(name: name, count: count,suffix: type == .weights ? "_W" : "_b",ext: "bin")
}if success {print(model.summary())
}

4、预测阶段：

model.encode(commandBuffer: commandBuffer, texture: inputTexture, inflightIndex: i)

let probabilities = model.outputImage(inflightIndex: i).toFloatArray()
let top5 = probabilities.top(k: 5)
let top5Labels = top5.map { x -> (String, Float) in (labels[x.0], x.1) }

深度学习（六十七）metal forge深度学习库使用相关推荐

问题六十七：ray tracing学习总结（2016.11.13, 2017.02.05）
从2016.11.13开始接触ray tracing到今天2017.02.05,差不多80天的时间.截至当前,学习ray tracing的过程,也是我重新找回自己或者说是"find what ...
springboot学习(六十七) springboot项目通过gradle-docker-plugin插件构建为doker镜像并推送至镜像私服
文章目录前言 1.使用gradle部署springboot项目 2.Docker开启远程访问 3.安装镜像私服Harbor 4.gradle中配置插件前言 springboot从2.4默认提供了打 ...
Docker学习六：综合实践
前言本次学习来自于datawhale组队学习: 教程地址为: https://github.com/datawhalechina/team-learning-program/tree/master/ ...
C1认证学习六（HTTPS）
C1认证学习六(HTTPS) 文章目录 C1认证学习六(HTTPS) 任务学习任务目标加密算法对称加密非对称加密不可逆加密 SSL TLS 任务学习 HTTPS是身披 SSL / TLS 的 ...
花书+吴恩达深度学习（十七）序列模型之长短期记忆 LSTM 和门控循环单元 GRU
目录 0. 前言 1. 长短期记忆 LSTM 2. 门控循环单元 GRU 如果这篇文章对你有一点小小的帮助,请给个关注,点个赞喔~我会非常开心的~ 花书+吴恩达深度学习(十五)序列模型之循环神经网络 ...
深度学习入门笔记（十七）：深度学习的极限在哪？
欢迎关注WX公众号:[程序员管小亮] 专栏--深度学习入门笔记声明 1)该文章整理自网上的大牛和机器学习专家无私奉献的资料,具体引用的资料请看参考文献. 2)本文仅供学术交流,非商用.所以每一部分具 ...
apple深度学习开发包Metal
转自:https://zhuanlan.zhihu.com/p/24623380?utm_source=tuicool&utm_medium=referral 今年的NIPS大会上,Googl ...
系统学习深度学习（十七）--VGG模型
原文:http://blog.csdn.net/u014114990/article/details/50715548 最近在看imagenet ILSVRC2015 比赛结果,除了 msra 的深 ...
深度强化学习控制六足机器人移动个人学习笔记（一）
深度强化学习控制六足机器人移动个人学习笔记(一) 注意事项 ubuntu18对应的ros版本不是Kinect gym算法执行中部分包要求Python不低于3.6 conda虚拟环境安装gym-gaze ...

深度学习（六十七）metal forge深度学习库使用

深度学习（六十七）metal forge深度学习库使用相关推荐

最新文章

热门文章