计算机视觉(斯坦福2017课程)
1. 主流深度学习框架
Caffe,Torch,Theano,Tensorflow,Keras,PyTorch等。
- 优势:
(1)框架能够让你轻松的构建和使用一个庞大的计算图
(2)能够轻松的在计算图中计算梯度
(3)能够高效的运行在GPU上
Caffe / Caffe2
Theano / TensorFlow
(1)写一些代码来定义计算图
(2)然后可以运行图模型,重复利用它
训练这个网络:
如上过程需要将权重不断的在CPU和GPU之间复制,因此一般不使用这种方式,而是将w也放在GPU处理。
代码如下:
Torch / PyTorch
AlexNet,VGG, GoogleNet,ResNet等